videocrafter / gradio_videolora.py
RamAnanth1's picture
Update gradio_videolora.py
c200095
raw
history blame
No virus
2.26 kB
import gradio as gr
lora_list = ["Loving Vincent","Frozen","MakotoShinkai YourName","Coco" ]
def create_demo(get_video_lora):
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)
with gr.Row(elem_id='prompt-container').style(equal_height=True):
model_choice = gr.Dropdown(choices=lora_list, value=lora_list[0], label='Model Style')
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,
model_choice
# num_frames,
# num_inference_steps,
]
gr.Examples(examples=[["A monkey is playing a piano", 1431,50, "Frozen"]],
inputs=inputs,
outputs=result,
fn=get_video_lora,
cache_examples=True)
prompt.submit(fn=get_video_lora, inputs=inputs, outputs=result)
run_button.click(fn=get_video_lora, inputs=inputs, outputs=result)
return block