Update app.py

app.py

@@ -101,7 +101,7 @@ def generate_video(prompt: str, frame1_path: str, frame2_path: str, guidance_sca
     cond_frame1 = Image.open(frame1_path)
     cond_frame2 = Image.open(frame2_path)
     
-    height, width = 720, 1280
+    height, width = 720, 720
     cond_frame1 = resize_image_to_bucket(cond_frame1, bucket_reso=(width, height))
     cond_frame2 = resize_image_to_bucket(cond_frame2, bucket_reso=(width, height))
     
@@ -199,9 +199,9 @@ def call_pipe(
     prompt: Union[str, List[str]] = None,
     prompt_2: Union[str, List[str]] = None,
     height: int = 720,
-    width: int = 1280,
+    width: int = 720,
     num_frames: int = 129,
-    num_inference_steps: int = 50,
+    num_inference_steps: int = 30,
     sigmas: Optional[List[float]] = None,
     guidance_scale: float = 6.0,
     num_videos_per_prompt: Optional[int] = 1,
@@ -346,6 +346,10 @@ def call_pipe(
     return HunyuanVideoPipelineOutput(frames=video)
 
 def main():
+    model_id = "hunyuanvideo-community/HunyuanVideo"
+    lora_path = hf_hub_download("dashtoon/hunyuan-video-keyframe-control-lora", "i2v.sft")  # Replace with the actual LORA path
+    pipe = construct_video_pipeline(model_id, lora_path)
+    
     gr.Markdown(
     """
     - https://i-bacon.bunkr.ru/11b45aa7-630b-4189-996f-a6b37a697786.png
@@ -367,10 +371,12 @@ def main():
     outputs = [
         gr.Video(label="Generated Video"),
     ]
-    
+    def generate_video_wrapper(*args):
+        return generate_video(pipe, *args)
+
     # Create the Gradio interface
     iface = gr.Interface(
-        fn=generate_video,
+        fn=generate_video_wrapper,
         inputs=inputs,
         outputs=outputs,
         title="Hunyuan Video Generator",