Update app.py

app.py

@@ -15,19 +15,41 @@ except Exception as e:
 os.environ["GRADIO_TEMP_DIR"] = "./tmp"
 
 try:
-    # You can define minimal dummy args here for initialization
+    # Define full dummy args with all attributes expected by load_models
     class DummyArgs:
+        # Core model paths
         wan_model_dir = "./models/Wan2.1-I2V-14B-720P"
         fantasytalking_model_path = "./models/fantasytalking_model.ckpt"
         wav2vec_model_dir = "./models/wav2vec2-base-960h"
 
+        # Required inference-related parameters
+        image_path = "./assets/images/woman.png"
+        audio_path = "./assets/audios/woman.wav"
+        prompt = "A woman is talking."
+        output_dir = "./output"
+        image_size = 512
+        audio_scale = 1.0
+        prompt_cfg_scale = 5.0
+        audio_cfg_scale = 5.0
+        max_num_frames = 81
+        inference_steps = 20
+        fps = 23
+        seed = 1111
+
+        # ✅ The missing one that caused your error:
+        num_persistent_param_in_dit = 7 * 10**9  # adjust if needed
+
+    # Preload models
+    print("🔄 Loading models into memory...")
     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
+    raise e  # fail fast if model load fails
+
 
 pipe,fantasytalking,wav2vec_processor,wav2vec = None,None,None,None
 @spaces.GPU(duration=1200)