|
import gradio as gr |
|
import subprocess |
|
from huggingface_hub import snapshot_download |
|
|
|
|
|
snapshot_download( |
|
repo_id = "Wan-AI/Wan2.1-T2V-1.3B", |
|
local_dir = "./Wan2.1-T2V-1.3B" |
|
) |
|
|
|
def infer(prompt): |
|
prompt = ""Two anthropomorphic cats in comfy boxing gear and bright gloves fight intensely on a spotlighted stage."" |
|
|
|
command = [ |
|
"python", "-m", "generate.py", |
|
"--task", "t2v-1.3B", |
|
"--size", "832*480", |
|
"--ckpt_dir", "./Wan2.1-T2V-1.3B", |
|
"--sample_shift", "8", |
|
"--sample_guide_scale", "6", |
|
"--prompt", f"{prompt}", |
|
"--save_file", "generated_video.mp4" |
|
] |
|
|
|
result = subprocess.run(command, capture_output=True, text=True) |
|
|
|
if result.returncode == 0: |
|
print("Command executed successfully.") |
|
return "./tmp/"generated_video.mp4"" |
|
else: |
|
print("Error executing command.") |
|
raise gr.Error("Error executing command") |
|
|
|
with gr.Blocks() as demo: |
|
with gr.Column(): |
|
gr.Markdown("# Wan 2.1") |
|
prompt = gr.Textbox(labe="Prompt") |
|
submit_btn = gr.Button("Submit") |
|
video_res = gr.Video(label="Generated Video") |
|
|
|
submit_btn.click( |
|
fn = infer, |
|
inputs = [prompt], |
|
outputs = [video_res] |
|
) |
|
|
|
demo.queue().launch(show_error=True, show_api=False, ssr_mode=False) |
