Update app.py

app.py

@@ -30,7 +30,7 @@ from torchao.quantization import Int8WeightOnlyConfig
 import aoti
 
 
-MODEL_ID = "alexander00001/NSFW_Wan_14b"
+MODEL_ID = "Wan2.2-I2V-A14B-Diffusers"
 
 MAX_DIM = 768
 MIN_DIM = 448
@@ -47,13 +47,13 @@ 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('alexander00001/NSFW_Wan_14b',
+    transformer=WanTransformer3DModel.from_pretrained('Wan2.2-I2V-A14B-Diffusers',
         subfolder='transformer',
         torch_dtype=torch.bfloat16,
         device_map='cuda',
     ),
-    transformer_2=WanTransformer3DModel.from_pretrained('alexander00001/NSFW_Wan_14b',
-        subfolder='transformer',
+    transformer_2=WanTransformer3DModel.from_pretrained('Wan2.2-I2V-A14B-Diffusers',
+        subfolder='transformer_2',
         torch_dtype=torch.bfloat16,
         device_map='cuda',
     ),