Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -52,7 +52,7 @@ def prepare_vertical_image(pipe, image, base_width=384, base_height=672):
|
|
| 52 |
# ================================
|
| 53 |
|
| 54 |
@spaces.GPU(size="xlarge", duration=180)
|
| 55 |
-
def generate_video(input_image, prompt, negative_prompt):
|
| 56 |
|
| 57 |
if input_image is None:
|
| 58 |
return None
|
|
@@ -69,7 +69,7 @@ def generate_video(input_image, prompt, negative_prompt):
|
|
| 69 |
width=width,
|
| 70 |
num_frames=161, # FIXED
|
| 71 |
guidance_scale=4.5,
|
| 72 |
-
num_inference_steps=
|
| 73 |
).frames[0]
|
| 74 |
|
| 75 |
output_path = "vertical_output.mp4"
|
|
|
|
| 52 |
# ================================
|
| 53 |
|
| 54 |
@spaces.GPU(size="xlarge", duration=180)
|
| 55 |
+
def generate_video(input_image, prompt, negative_prompt, progress=gr.Progress(track_tqdm=True)):
|
| 56 |
|
| 57 |
if input_image is None:
|
| 58 |
return None
|
|
|
|
| 69 |
width=width,
|
| 70 |
num_frames=161, # FIXED
|
| 71 |
guidance_scale=4.5,
|
| 72 |
+
num_inference_steps=10
|
| 73 |
).frames[0]
|
| 74 |
|
| 75 |
output_path = "vertical_output.mp4"
|