Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -43,6 +43,8 @@ def generate_image(prompt, base="Realistic", motion="", step=8, progress=gr.Prog
|
|
| 43 |
global motion_loaded
|
| 44 |
print(prompt, base, step)
|
| 45 |
|
|
|
|
|
|
|
| 46 |
if step_loaded != step:
|
| 47 |
repo = "ByteDance/AnimateDiff-Lightning"
|
| 48 |
ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
|
|
@@ -155,4 +157,4 @@ with gr.Blocks(css=css) as demo:
|
|
| 155 |
elem_id="gr-examples"
|
| 156 |
)
|
| 157 |
|
| 158 |
-
demo.queue(
|
|
|
|
| 43 |
global motion_loaded
|
| 44 |
print(prompt, base, step)
|
| 45 |
|
| 46 |
+
step = int(step)
|
| 47 |
+
|
| 48 |
if step_loaded != step:
|
| 49 |
repo = "ByteDance/AnimateDiff-Lightning"
|
| 50 |
ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
|
|
|
|
| 157 |
elem_id="gr-examples"
|
| 158 |
)
|
| 159 |
|
| 160 |
+
demo.queue(show_error=True).launch()
|