multimodalart HF Staff commited on
Commit
ee6ffaa
·
verified ·
1 Parent(s): 7c62e59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -102,7 +102,7 @@ def calculate_dimensions(width, height):
102
  return (new_width // 16) * 16, (new_height // 16) * 16
103
 
104
  @spaces.GPU(duration=120)
105
- def generate_video(prompt, first_image_path, last_image_path, duration_seconds):
106
  # Create a temporary directory for resized images
107
  temp_dir = "input"
108
  os.makedirs(temp_dir, exist_ok=True)
@@ -156,7 +156,7 @@ def generate_video(prompt, first_image_path, last_image_path, duration_seconds):
156
 
157
  vaedecode_8 = vaedecode.decode(samples=get_value_at_index(ksampleradvanced_102, 0), vae=get_value_at_index(vaeloader_39, 0))
158
  createvideo_104 = createvideo.create_video(fps=16, images=get_value_at_index(vaedecode_8, 0))
159
- savevideo_103 = savevideo.save_video(filename_prefix="ComfyUI_Video", format="mp4", codec="libx264", video=get_value_at_index(createvideo_104, 0))
160
 
161
  return f"output/{savevideo_103['ui']['videos'][0]['filename']}"
162
 
 
102
  return (new_width // 16) * 16, (new_height // 16) * 16
103
 
104
  @spaces.GPU(duration=120)
105
+ def generate_video(prompt, first_image_path, last_image_path, duration_seconds, progress=gr.Progress(track_tqdm=True)):
106
  # Create a temporary directory for resized images
107
  temp_dir = "input"
108
  os.makedirs(temp_dir, exist_ok=True)
 
156
 
157
  vaedecode_8 = vaedecode.decode(samples=get_value_at_index(ksampleradvanced_102, 0), vae=get_value_at_index(vaeloader_39, 0))
158
  createvideo_104 = createvideo.create_video(fps=16, images=get_value_at_index(vaedecode_8, 0))
159
+ savevideo_103 = savevideo.save_video(filename_prefix="ComfyUI_Video", format="mp4", codec="h264", video=get_value_at_index(createvideo_104, 0))
160
 
161
  return f"output/{savevideo_103['ui']['videos'][0]['filename']}"
162