Update app.py

app.py

@@ -5,7 +5,7 @@ os.system('pip install --upgrade --pre --extra-index-url https://download.pytorc
 # Actual demo code
 import spaces
 import torch
-from diffusers.pipelines.wan.pipeline_wan_i2v import WanImageToVideoPipeline
+from diffusers.pipelines.wan.pipeline_wan_i2v import WanTextToVideoPipeline
 from diffusers.models.transformers.transformer_wan import WanTransformer3DModel
 from diffusers.utils.export_utils import export_to_video
 import gradio as gr
@@ -31,7 +31,7 @@ MIN_DURATION = round(MIN_FRAMES_MODEL/FIXED_FPS,1)
 MAX_DURATION = round(MAX_FRAMES_MODEL/FIXED_FPS,1)
 
 
-pipe = WanImageToVideoPipeline.from_pretrained(MODEL_ID,
+pipe = WanTextToVideoPipeline.from_pretrained(MODEL_ID,
     transformer=WanTransformer3DModel.from_pretrained('cbensimon/Wan2.2-I2V-A14B-bf16-Diffusers',
         subfolder='transformer',
         torch_dtype=torch.bfloat16,