Spaces:
Runtime error
Runtime error
new changes
#4
by
amoljk
- opened
app.py
CHANGED
|
@@ -71,11 +71,6 @@ def generate_image(prompt, base, motion, step, progress=gr.Progress()):
|
|
| 71 |
|
| 72 |
output = pipe(prompt=prompt, guidance_scale=1.0, num_inference_steps=step, callback=progress_callback, callback_steps=1)
|
| 73 |
|
| 74 |
-
has_nsfw_concepts = check_nsfw_images([output.frames[0][0]])
|
| 75 |
-
if has_nsfw_concepts[0]:
|
| 76 |
-
gr.Warning("NSFW content detected.")
|
| 77 |
-
return None
|
| 78 |
-
|
| 79 |
name = str(uuid.uuid4()).replace("-", "")
|
| 80 |
path = f"/tmp/{name}.mp4"
|
| 81 |
export_to_video(output.frames[0], path, fps=10)
|
|
|
|
| 71 |
|
| 72 |
output = pipe(prompt=prompt, guidance_scale=1.0, num_inference_steps=step, callback=progress_callback, callback_steps=1)
|
| 73 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 74 |
name = str(uuid.uuid4()).replace("-", "")
|
| 75 |
path = f"/tmp/{name}.mp4"
|
| 76 |
export_to_video(output.frames[0], path, fps=10)
|