Add examples
Browse files- 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],
|