Update app.py

app.py

@@ -15,16 +15,19 @@ except Exception as e:
 os.environ["GRADIO_TEMP_DIR"] = "./tmp"
 
 try:
-        global pipe, fantasytalking, wav2vec_processor, wav2vec  
-        
-        pipe,fantasytalking,wav2vec_processor,wav2vec = load_models(args)
-        output_path=main(
-            args,pipe,fantasytalking,wav2vec_processor,wav2vec
-        )
-        return output_path  # Ensure the output path is returned
-    except Exception as e:
-        print(f"Error during processing: {str(e)}")
-        raise gr.Error(f"Error during processing: {str(e)}")
+    # You can define minimal dummy args here for initialization
+    class DummyArgs:
+        wan_model_dir = "./models/Wan2.1-I2V-14B-720P"
+        fantasytalking_model_path = "./models/fantasytalking_model.ckpt"
+        wav2vec_model_dir = "./models/wav2vec2-base-960h"
+
+    args = DummyArgs()
+    pipe, fantasytalking, wav2vec_processor, wav2vec = load_models(args)
+    print("✅ Models loaded successfully.")
+except Exception as e:
+    print(f"❌ Error loading models: {e}")
+    pipe = fantasytalking = wav2vec_processor = wav2vec = None
+    raise e  # fail fast on startup if models can't load
 
 pipe,fantasytalking,wav2vec_processor,wav2vec = None,None,None,None
 @spaces.GPU(duration=1200)