Maximofn commited on
Commit
e07fcea
·
1 Parent(s): ce24f93

Modify load_example_video function to accept variable arguments

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -115,7 +115,7 @@ def generate_video(transformer_model, prompt, negative_prompt, num_frames, heigh
115
  print("Saving video")
116
  return save_video(video_frames)
117
 
118
- def load_example_video(example_index):
119
  """Carga un video de ejemplo desde la carpeta example_outputs"""
120
  example_path = f"example_outputs/example_{example_index}.mp4"
121
  if os.path.exists(example_path):
 
115
  print("Saving video")
116
  return save_video(video_frames)
117
 
118
+ def load_example_video(example_index, *args):
119
  """Carga un video de ejemplo desde la carpeta example_outputs"""
120
  example_path = f"example_outputs/example_{example_index}.mp4"
121
  if os.path.exists(example_path):