Update app.py

app.py

@@ -40,11 +40,7 @@ try:
         raise FileNotFoundError(f"Base model not found at: {BASE_MODEL_PATH}")
     if not os.path.exists(LORA_MODEL_PATH):
         raise FileNotFoundError(f"LoRA model not found at: {LORA_MODEL_PATH}")
-
-    face_processor = FaceProcessor(antelopv2_path=ANTELOPEV2_PATH)
-    pipe = load_wan_pipe(base_path=BASE_MODEL_PATH, torch_dtype=torch.bfloat16)
-    set_stand_in(pipe, model_path=LORA_MODEL_PATH)
-    print("Model loaded successfully!")
+ 
 except Exception as e:
     print(f"Model loading failed: {e}")
     with gr.Blocks() as demo:
@@ -74,6 +70,11 @@ def generate_video(
     if pil_image is None:
         raise gr.Error("Please upload a face image first!")
 
+    face_processor = FaceProcessor(antelopv2_path=ANTELOPEV2_PATH)
+    pipe = load_wan_pipe(base_path=BASE_MODEL_PATH, torch_dtype=torch.bfloat16)
+    set_stand_in(pipe, model_path=LORA_MODEL_PATH)
+    print("Model loaded successfully!")
+
     print("Processing face...")
     ip_image = face_processor.process(pil_image)
     print("Face processing completed.")