Spaces:
Runtime error
Runtime error
Update
Browse files
app.py
CHANGED
|
@@ -30,7 +30,7 @@ pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, times
|
|
| 30 |
|
| 31 |
# Function
|
| 32 |
@spaces.GPU(enable_queue=True)
|
| 33 |
-
def generate_image(prompt, base, motion, step):
|
| 34 |
global step_loaded
|
| 35 |
global base_loaded
|
| 36 |
global motion_loaded
|
|
@@ -53,7 +53,11 @@ def generate_image(prompt, base, motion, step):
|
|
| 53 |
pipe.set_adapters(["motion"], [0.7])
|
| 54 |
motion_loaded = motion
|
| 55 |
|
| 56 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 57 |
name = str(uuid.uuid4()).replace("-", "")
|
| 58 |
path = f"/tmp/{name}.mp4"
|
| 59 |
export_to_video(output.frames[0], path, fps=10)
|
|
|
|
| 30 |
|
| 31 |
# Function
|
| 32 |
@spaces.GPU(enable_queue=True)
|
| 33 |
+
def generate_image(prompt, base, motion, step, progress=gr.Progress()):
|
| 34 |
global step_loaded
|
| 35 |
global base_loaded
|
| 36 |
global motion_loaded
|
|
|
|
| 53 |
pipe.set_adapters(["motion"], [0.7])
|
| 54 |
motion_loaded = motion
|
| 55 |
|
| 56 |
+
progress((0, step))
|
| 57 |
+
def progress_callback(i, t, z):
|
| 58 |
+
progress((i+1, step))
|
| 59 |
+
|
| 60 |
+
output = pipe(prompt=prompt, guidance_scale=1.0, num_inference_steps=step, callback=progress_callback)
|
| 61 |
name = str(uuid.uuid4()).replace("-", "")
|
| 62 |
path = f"/tmp/{name}.mp4"
|
| 63 |
export_to_video(output.frames[0], path, fps=10)
|