videocrafter / gradio_t2v.py
RamAnanth1's picture
Update gradio_t2v.py
b286159
import gradio as gr
def create_demo(get_video):
block = gr.Blocks(css='style.css').queue()
with block:
with gr.Group():
with gr.Box():
with gr.Row(elem_id='prompt-container').style(equal_height=True):
prompt = gr.Text(
label='Prompt',
show_label=False,
max_lines=1,
placeholder='Enter your prompt',
elem_id='prompt-text-input').style(container=False)
run_button = gr.Button('Generate video').style(
full_width=False)
result = gr.Video(label='Result', show_label=False, elem_id='gallery')
with gr.Accordion('Advanced options', open=False):
seed = gr.Slider(
label='Seed',
minimum=-1,
maximum=1000000,
step=1,
value=-1,
info='If set to -1, a different seed will be used each time.')
sampling_steps = gr.Slider(label='Number of sampling steps',
minimum=10,
maximum=100,
step=5,
value=50)
inputs = [
prompt,
seed,
sampling_steps
# num_frames,
# num_inference_steps,
]
gr.Examples(examples=[
["Astronaut riding a horse", 731, 50],
["Cars running on the highway at night", 1000, 50],
["close up of a clown fish swimming. 4K",865, 50]
],
inputs=inputs,
outputs=result,
fn=get_video,
cache_examples=True)
prompt.submit(fn=get_video, inputs=inputs, outputs=result)
run_button.click(fn=get_video, inputs=inputs, outputs=result)
return block