Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import sys
+from diffusers import WanImageToVideoPipeline, AutoencoderKLWan, ModularPi
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
 
 #import subprocess
@@ -31,7 +32,7 @@ ckpt_dir = snapshot_download(repo_id, local_dir_use_symlinks=False)
 print(f"Using checkpoints from {ckpt_dir}")
 
 # Load the model configuration
-TASK_NAME = 'ti2v-5B'
+TASK_NAME = 'ti2v-5B-Diff'
 cfg = WAN_CONFIGS[TASK_NAME]
 FIXED_FPS = 24
 MIN_FRAMES_MODEL = 8
@@ -41,6 +42,10 @@ MAX_FRAMES_MODEL = 121
 print("Initializing WanTI2V pipeline...")
 device = "cuda" if torch.cuda.is_available() else "cpu"
 device_id = 0 if torch.cuda.is_available() else -1
+
+
+
+
 pipeline = wan.WanTI2V(
     config=cfg,
     checkpoint_dir=ckpt_dir,