Update app.py

app.py

@@ -1,29 +1,12 @@
 import torch
-import gradio as gr
-from diffusers import StableDiffusionPipeline
+from diffusers import WanPipeline
+from diffusers.utils import export_to_video
 
-# Model path from Hugging Face
-MODEL_NAME = "Evados/DiffSynth-Studio-Lora-Wan2.1-ComfyUI"
-LORA_FILE = "Wan2.1-1.3b-lora-aesthetics-v1_new.safetensors"
+pipe = WanPipeline.from_pretrained(
+    "Wan-AI/Wan2.1-T2V-1.3B-Diffusers", torch_dtype=torch.bfloat16
+).to("cuda")
+pipe.load_lora_weights("finetrainers/Wan2.1-T2V-1.3B-3dgs-v0", adapter_name="wan-lora")
+pipe.set_adapters(["wan-lora"], [0.5])
 
-def load_model():
-    pipe = StableDiffusionPipeline.from_pretrained(MODEL_NAME, torch_dtype=torch.float16).to("cuda")
-    pipe.load_lora_weights(f"{MODEL_NAME}/{LORA_FILE}")
-    return pipe
-
-pipe = load_model()
-
-def generate_image(prompt):
-    image = pipe(prompt).images[0]
-    return image
-
-# Gradio UI
-iface = gr.Interface(
-    fn=generate_image,
-    inputs=gr.Textbox(label="Enter your prompt"),
-    outputs=gr.Image(label="Generated Image"),
-    title="Wan2.1 LoRA Image Generator",
-    description="Generate images using the Wan2.1 LoRA model. Enter a prompt to begin."
-)
-
-iface.launch()
+video = pipe("<my-awesome-prompt>").frames[0]
+export_to_video(video, "output.mp4", fps=24)