Update app.py

app.py

@@ -1,5 +1,5 @@
 import torch
-from diffusers import AutoencoderKLWan, WanTextToVideoPipeline, UniPCMultistepScheduler
+from diffusers import AutoencoderKLWan, WanPipeline, UniPCMultistepScheduler
 from diffusers.utils import export_to_video
 import gradio as gr
 import tempfile
@@ -13,7 +13,7 @@ LORA_REPO_ID = "Kijai/WanVideo_comfy"
 LORA_FILENAME = "Wan21_CausVid_bidirect2_T2V_1_3B_lora_rank32.safetensors"
 
 vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)
-pipe = WanTextToVideoPipeline.from_pretrained(
+pipe = WanPipeline.from_pretrained(
     MODEL_ID, vae=vae, torch_dtype=torch.bfloat16
 )
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=8.0)