Add badges

gradio_app.py

@@ -178,7 +178,19 @@ with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# Allegro Video Generation")
         gr.Markdown("Generate a video based on a text prompt using the Allegro pipeline.")
-        
+        gr.HTML("""
+        <div style="display:flex;column-gap:4px;">
+            <a href='https://huggingface.co/rhymes-ai/Allegro'>
+                <img src='https://img.shields.io/badge/HuggingFace-Model-orange'>
+            </a> 
+            <a href='https://github.com/rhymes-ai/Allegro/tree/main'>
+                <img src='https://img.shields.io/badge/GitHub-Repo-blue'>
+            </a> 
+            <a href='https://arxiv.org/abs/2410.15458'>
+                <img src='https://img.shields.io/badge/ArXivPaper-red'>
+            </a>
+        </div>
+        """)
         user_prompt=gr.Textbox(label="User Prompt")
         with gr.Row():
             guidance_scale=gr.Slider(minimum=0, maximum=20, step=0.1, label="Guidance Scale", value=7.5)