Update app.py

app.py

@@ -51,7 +51,7 @@ except Exception as e:
     raise e  # fail fast if model load fails
 
 
-pipe,fantasytalking,wav2vec_processor,wav2vec = None,None,None,None
+# pipe,fantasytalking,wav2vec_processor,wav2vec = None,None,None,None
 @spaces.GPU(duration=1200)
 def generate_video(
     image_path,
@@ -65,30 +65,35 @@ def generate_video(
     inference_steps,
     seed,
 ):
-    # Create the temp directory if it doesn't exist
-    output_dir = Path("./output")
-    output_dir.mkdir(parents=True, exist_ok=True)
+    try:
+        output_dir = Path("./output")
+        output_dir.mkdir(parents=True, exist_ok=True)
 
-    # Convert paths to absolute Path objects and normalize them
-    print(image_path)
-    image_path = Path(image_path).absolute().as_posix()
-    audio_path = Path(audio_path).absolute().as_posix()
+        image_path = Path(image_path).absolute().as_posix()
+        audio_path = Path(audio_path).absolute().as_posix()
 
-    # Parse the arguments
+        args = create_args(
+            image_path=image_path,
+            audio_path=audio_path,
+            prompt=prompt or "A person is talking.",
+            output_dir=str(output_dir),
+            audio_weight=audio_weight,
+            prompt_cfg_scale=prompt_cfg_scale,
+            audio_cfg_scale=audio_cfg_scale,
+            image_size=image_size,
+            max_num_frames=max_num_frames,
+            inference_steps=inference_steps,
+            seed=seed,
+        )
 
-    args = create_args(
-        image_path=image_path,
-        audio_path=audio_path,
-        prompt=prompt,
-        output_dir=str(output_dir),
-        audio_weight=audio_weight,
-        prompt_cfg_scale=prompt_cfg_scale,
-        audio_cfg_scale=audio_cfg_scale,
-        image_size=image_size,
-        max_num_frames=max_num_frames,
-        inference_steps=inference_steps,
-        seed=seed,
-    )
+        # ✅ Run inference using preloaded models
+        save_path = main(args, pipe, fantasytalking, wav2vec_processor, wav2vec)
+        print(f"✅ Video saved at {save_path}")
+        return save_path
+
+    except Exception as e:
+        print(f"❌ Error generating video: {e}")
+        return None