Spaces:
Running
on
Zero
Running
on
Zero
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)
|