Ujwaltheja commited on
Commit
a079a65
·
verified ·
1 Parent(s): d1b6015

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -0
app.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import torch
3
+ from diffusers import DiffusionPipeline
4
+ import imageio
5
+ import tempfile
6
+
7
+ # Load model (first run will download weights)
8
+ pipe = DiffusionPipeline.from_pretrained(
9
+ "Wan-AI/Wan2.2-T2V-A14B",
10
+ torch_dtype=torch.float16
11
+ ).to("cuda")
12
+
13
+ def generate_video(prompt, steps=25):
14
+ video = pipe(prompt, num_inference_steps=steps).videos[0]
15
+ tmpfile = tempfile.NamedTemporaryFile(suffix=".gif", delete=False)
16
+ imageio.mimsave(tmpfile.name, video, fps=8)
17
+ return tmpfile.name
18
+
19
+ demo = gr.Interface(
20
+ fn=generate_video,
21
+ inputs=[gr.Textbox(label="Prompt"), gr.Slider(1, 50, value=25, label="Steps")],
22
+ outputs=gr.Video(label="Generated Video")
23
+ )
24
+
25
+ demo.launch()