Sushantkas commited on
Commit
fc67fc0
·
verified ·
1 Parent(s): 5820239

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=25
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"