RamAnanth1 commited on
Commit
63526ee
β€’
1 Parent(s): f7f313b

Update gradio_videolora.py

Browse files
Files changed (1) hide show
  1. gradio_videolora.py +6 -2
gradio_videolora.py CHANGED
@@ -1,5 +1,7 @@
1
  import gradio as gr
2
 
 
 
3
  def create_demo(get_video_lora):
4
  block = gr.Blocks(css='style.css').queue()
5
  with block:
@@ -12,6 +14,7 @@ def create_demo(get_video_lora):
12
  max_lines=1,
13
  placeholder='Enter your prompt',
14
  elem_id='prompt-text-input').style(container=False)
 
15
  run_button = gr.Button('Generate video').style(
16
  full_width=False)
17
  result = gr.Video(label='Result', show_label=False, elem_id='gallery')
@@ -32,11 +35,12 @@ def create_demo(get_video_lora):
32
  inputs = [
33
  prompt,
34
  seed,
35
- sampling_steps
 
36
  # num_frames,
37
  # num_inference_steps,
38
  ]
39
- gr.Examples(examples=[["A monkey is playing a piano, frozenmovie style", 431,50]],
40
  inputs=inputs,
41
  outputs=result,
42
  fn=get_video_lora,
 
1
  import gradio as gr
2
 
3
+ lora_list = ["Loving Vincent","Frozen","MakotoShinkai YourName","Coco" ]
4
+
5
  def create_demo(get_video_lora):
6
  block = gr.Blocks(css='style.css').queue()
7
  with block:
 
14
  max_lines=1,
15
  placeholder='Enter your prompt',
16
  elem_id='prompt-text-input').style(container=False)
17
+ model_choice = gr.Dropdown(choices=lora_list, value=lora_list[0], label='Model Style')
18
  run_button = gr.Button('Generate video').style(
19
  full_width=False)
20
  result = gr.Video(label='Result', show_label=False, elem_id='gallery')
 
35
  inputs = [
36
  prompt,
37
  seed,
38
+ sampling_steps,
39
+ model_choice
40
  # num_frames,
41
  # num_inference_steps,
42
  ]
43
+ gr.Examples(examples=[["A monkey is playing a piano", 1431,50, "Frozen"]],
44
  inputs=inputs,
45
  outputs=result,
46
  fn=get_video_lora,