fffiloni commited on
Commit
d63ddc4
1 Parent(s): eef9e5e

Add examples

Browse files
Files changed (1) hide show
  1. gradio_app.py +20 -0
gradio_app.py CHANGED
@@ -244,6 +244,26 @@ with gr.Blocks(css=css) as demo:
244
 
245
  video_output=gr.Video(label="Generated Video")
246
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
247
  submit_btn.click(
248
  fn=run_inference,
249
  inputs=[user_prompt, guidance_scale, num_sampling_steps, seed, enable_cpu_offload],
 
244
 
245
  video_output=gr.Video(label="Generated Video")
246
 
247
+ def load_allegro_examples(prompt):
248
+ if prompt == "A Monkey is playing bass guitar.":
249
+ return "https://rhymes.ai/allegroVideos/30_demo_w_watermark_prompt_1018/11.mp4"
250
+ elif prompt == "An astronaut riding a horde.":
251
+ return "https://rhymes.ai/allegroVideos/30_demo_w_watermark_prompt_1018/15.mp4"
252
+ elif prompt == "A tiny finch on a branch with spring flowers on background.":
253
+ return "https://rhymes.ai/allegroVideos/30_demo_w_watermark_prompt_1018/22.mp4"
254
+
255
+ gr.Examples(
256
+ examples=[
257
+ ["A Monkey is playing bass guitar."],
258
+ ["An astronaut riding a horse."],
259
+ ["A tiny finch on a branch with spring flowers on background."]
260
+ ],
261
+ fn=load_allegro_examples,
262
+ inputs=[user_prompt],
263
+ outputs=video_output,
264
+ run_on_click=True
265
+ )
266
+
267
  submit_btn.click(
268
  fn=run_inference,
269
  inputs=[user_prompt, guidance_scale, num_sampling_steps, seed, enable_cpu_offload],