Update app.py

app.py

@@ -4,9 +4,9 @@ from diffusers import DiffusionPipeline
 import imageio
 import tempfile
 
-# Load model (first run will download weights)
+# Load the lighter 5B model
 pipe = DiffusionPipeline.from_pretrained(
-    "Wan-AI/Wan2.2-T2V-A14B",
+    "Wan-AI/Wan2.2-TI2V-5B-Diffusers",
     torch_dtype=torch.float16
 ).to("cuda")
 
@@ -22,4 +22,5 @@ demo = gr.Interface(
     outputs=gr.Video(label="Generated Video")
 )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()