|
import gradio as gr |
|
import torch |
|
from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler |
|
from diffusers.utils import export_to_video |
|
|
|
pipe = DiffusionPipeline.from_pretrained("cerspense/zeroscope_v2_576w", torch_dtype=torch.float16) |
|
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config) |
|
pipe.enable_model_cpu_offload() |
|
|
|
def infer(prompt, num_inference_steps): |
|
|
|
video_frames = pipe(prompt, num_inference_steps=40, height=320, width=576, num_frames=24).frames |
|
video_path = export_to_video(video_frames) |
|
print(video_path) |
|
return video_path |
|
|
|
css = """ |
|
#col-container {max-width: 510px; margin-left: auto; margin-right: auto;} |
|
a {text-decoration-line: underline; font-weight: 600;} |
|
""" |
|
|
|
with gr.Blocks(css=css) as demo: |
|
with gr.Column(elem_id="col-container"): |
|
gr.HTML("""<div style="text-align: center; max-width: 700px; margin: 0 auto;"> |
|
<div |
|
style=" |
|
display: inline-flex; |
|
align-items: center; |
|
gap: 0.8rem; |
|
font-size: 1.75rem; |
|
" |
|
> |
|
<h1 style="font-weight: 900; margin-bottom: 7px; margin-top: 5px;"> |
|
Zeroscope Text-to-Video |
|
</h1> |
|
</div> |
|
<p style="margin-bottom: 10px; font-size: 94%"> |
|
A watermark-free Modelscope-based video model optimized for producing high-quality 16:9 compositions and a smooth video output. <br /> |
|
This model was trained using 9,923 clips and 29,769 tagged frames at 24 frames, 576x320 resolution. |
|
|
|
</p> |
|
</div>""") |
|
|
|
prompt_in = gr.Textbox(label="Prompt", placeholder="Darth Vader is surfing on waves") |
|
inference_steps = gr.Slider(minimum=10, maximum=100, step=1, value=40, interactive=False) |
|
submit_btn = gr.Button("Submit") |
|
video_result = gr.Video(label="Video Output") |
|
|
|
submit_btn.click(fn=infer, |
|
inputs=[prompt_in, inference_steps], |
|
outputs=[video_result]) |
|
|
|
demo.queue(max_size=12).launch() |
|
|