Update app.py

app.py

@@ -3,8 +3,13 @@ import torch
 from diffusers import DiffusionPipeline
 
 # Load the WAN 2.1 T2V Model
-device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = DiffusionPipeline.from_pretrained("Isi99999/Wan2.1-T2V-1.3B").to(device)
+
+from diffusers import DiffusionPipeline
+
+pipe = DiffusionPipeline.from_pretrained("StevenZhang/Wan2.1-T2V-1.3B-Diff")
+
+prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
+image = pipe(prompt).images[0]
 
 def generate_image(prompt):
     """Generates an image from text prompt using WAN 2.1"""