LPX55 commited on
Commit
8d52ba2
·
verified ·
1 Parent(s): 141fd6e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -96,7 +96,7 @@ def construct_video_pipeline(model_id: str, lora_path: str):
96
  pipe.unload_lora_weights()
97
 
98
  return pipe
99
- def generate_video(prompt: str, frame1_path: str, frame2_path: str, guidance_scale: float, num_frames: int, num_inference_steps: int) -> bytes:
100
  # Load and preprocess frames
101
  cond_frame1 = Image.open(frame1_path)
102
  cond_frame2 = Image.open(frame2_path)
 
96
  pipe.unload_lora_weights()
97
 
98
  return pipe
99
+ def generate_video(pipe, prompt: str, frame1_path: str, frame2_path: str, guidance_scale: float, num_frames: int, num_inference_steps: int) -> bytes:
100
  # Load and preprocess frames
101
  cond_frame1 = Image.open(frame1_path)
102
  cond_frame2 = Image.open(frame2_path)