Update app.py

app.py

@@ -21,18 +21,18 @@ os.environ['HF_HUB_CACHE'] = '/tmp/hub'  # Use temp directory to avoid filling p
 
 from diffusers import UniPCMultistepScheduler
 from transformers import CLIPTextModel, CLIPTokenizer
-from wan_diffusers import WanTextToVideoPipeline  # Use correct import for Wan's T2V pipeline
+
 from safetensors.torch import load_file
 from huggingface_hub import hf_hub_download
 import torch
 
 # --- Base model setup (Wan T2V) ---
-MODEL_ID = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
-LORA_FILENAME = "FusionX_LoRa/Wan2.1_T2V_14B_FusionX_LoRA.safetensors"
+MODEL_ID = "Wan-AI/Wan2.1-I2V-14B-Diffusers"
+LORA_FILENAME = "FusionX_LoRa/Wan2.1_I2V_14B_FusionX_LoRA.safetensors"
 
 print("🚀 Loading Wan2.1 T2V base pipeline...")
 
-pipe = WanTextToVideoPipeline.from_pretrained(
+pipe = WanImageToVideoPipeline.from_pretrained(
     MODEL_ID,
     torch_dtype=torch.bfloat16,
 )