Wan2GP_Demo / app.py
USF00's picture
Create app.py
438e50a verified
import gradio as gr
import subprocess
def generate_video(prompt, steps=25, frames=24):
cmd = [
"python", "wgp.py",
"--t2v",
"--fp16",
"--advanced",
"--prompt", prompt,
"--frames", str(frames),
"--steps", str(steps),
"--output", "generated.mp4"
]
try:
subprocess.run(cmd, check=True)
return "generated.mp4"
except Exception as e:
return str(e)
demo = gr.Interface(
fn=generate_video,
inputs=[
gr.Textbox(label="Prompt", value="A boy reading a book under a tree, cinematic"),
gr.Slider(minimum=5, maximum=60, value=25, step=1, label="Steps"),
gr.Slider(minimum=16, maximum=80, value=24, step=1, label="Frames")
],
outputs=gr.Video(label="Generated Video")
)
if __name__ == "__main__":
demo.launch()