Spaces:
Running
Running
File size: 2,319 Bytes
714bf26 2d7762b d76fbc2 714bf26 2d7762b 714bf26 2d7762b 714bf26 2d7762b 714bf26 7fe2981 714bf26 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
import gradio as gr
from model import Model
from functools import partial
import os
examples = [
["an astronaut waving the arm on the moon"],
["a sloth surfing on a wakeboard"],
["an astronaut walking on a street"],
["a cute cat walking on grass"],
["a horse is galloping on a street"],
["an astronaut is skiing down the hill"],
["a gorilla walking alone down the street"],
["a gorilla dancing on times square"],
["A panda dancing dancing like crazy on Times Square"],
]
def create_demo(model: Model):
with gr.Blocks() as demo:
with gr.Row():
gr.Markdown('## Text2Video-Zero: Video Generation')
with gr.Row():
with gr.Column():
prompt = gr.Textbox(label='Prompt')
run_button = gr.Button(label='Run')
with gr.Accordion('Advanced options', open=False):
motion_field_strength_x = gr.Slider(label='Global Translation $\delta_{x}$',
minimum=-20,
maximum=20,
value=12,
step=1)
motion_field_strength_y = gr.Slider(label='Global Translation $\delta_{y}$',
minimum=-20,
maximum=20,
value=12,
step=1)
# a_prompt = gr.Textbox(label="Added Prompt", value='best quality, extremely detailed')
n_prompt = gr.Textbox(label="Optional Negative Prompt",
value='')
with gr.Column():
result = gr.Video(label="Generated Video")
inputs = [
prompt,
motion_field_strength_x,
motion_field_strength_y,
n_prompt
]
gr.Examples(examples=examples,
inputs=inputs,
outputs=result,
# cache_examples=os.getenv('SYSTEM') == 'spaces',
run_on_click=False,
)
run_button.click(fn=model.process_text2video,
inputs=inputs,
outputs=result,)
return demo
|