Update app.py

app.py

@@ -30,7 +30,7 @@ from torchao.quantization import Int8WeightOnlyConfig
 import aoti
 
 
-MODEL_ID = "Wan2.2-I2V-A14B-Diffusers"
+MODEL_ID = "Wan-AI/Wan2.2-I2V-A14B-Diffusers"
 
 MAX_DIM = 768
 MIN_DIM = 448
@@ -47,12 +47,12 @@ MIN_DURATION = round(MIN_FRAMES_MODEL/FIXED_FPS,1)
 MAX_DURATION = round(MAX_FRAMES_MODEL/FIXED_FPS,1)
 
 pipe = WanImageToVideoPipeline.from_pretrained(MODEL_ID,
-    transformer=WanTransformer3DModel.from_pretrained('Wan2.2-I2V-A14B-Diffusers',
+    transformer=WanTransformer3DModel.from_pretrained('Wan-AI/Wan2.2-I2V-A14B-Diffusers',
         subfolder='transformer',
         torch_dtype=torch.bfloat16,
         device_map='cuda',
     ),
-    transformer_2=WanTransformer3DModel.from_pretrained('Wan2.2-I2V-A14B-Diffusers',
+    transformer_2=WanTransformer3DModel.from_pretrained('Wan-AI/Wan2.2-I2V-A14B-Diffusers',
         subfolder='transformer_2',
         torch_dtype=torch.bfloat16,
         device_map='cuda',