add examples + credits

app.py

@@ -5,7 +5,7 @@ from auffusion_pipeline import AuffusionPipeline
 
 pipeline = AuffusionPipeline.from_pretrained("auffusion/auffusion")
 
-def infer(prompt):
+def infer(prompt, progress=gr.Progress(track_tqdm=True)):
     
     prompt = prompt
     output = pipeline(prompt=prompt)
@@ -14,13 +14,45 @@ def infer(prompt):
 
     return f"{prompt}.wav"
 
-with gr.Blocks() as demo:
-    with gr.Column():
+css="""
+div#col-container{
+    margin: 0 auto;
+    max-width: 640px;
+}
+"""
+with gr.Blocks(css=css) as demo:
+    with gr.Column(elem_id="col-container"):
         gr.Markdown("# Auffusion")
+        gr.Markdown("Auffusion can generate realistic audios including human sounds, animal sounds, natural and artificial sounds and sound effects from textual prompts. ")
+        gr.HTML("""
+        <div style="display:flex;column-gap:4px;">
+            <a href="https://auffusion.github.io/">
+                <img src='https://img.shields.io/badge/Project-Page-green'>
+            </a>
+            <a href="https://github.com/happylittlecat2333/Auffusion">
+                <img src='https://img.shields.io/badge/GitHub-Repo-blue'>
+            </a> 
+            <a href="https://arxiv.org/pdf/2401.01044">
+                <img src='https://img.shields.io/badge/ArXiv-Paper-red'>
+            </a>
+            <a href="https://huggingface.co/spaces/fffiloni/auffusion?duplicate=true">
+                <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-sm.svg" alt="Duplicate this Space">
+            </a>
+        </div>
+        """)
+        
         prompt = gr.Textbox(label="Prompt")
         submit_btn = gr.Button("Submit")
         audio_out = gr.Audio(label="Audio Ressult")
 
+        gr.Examples(
+            examples = [
+                "Rolling thunder with lightning strikes",
+                "Two gunshots followed by birds chirping",
+                "A train whistle blowing in the distance"
+            ]
+        )
+
     submit_btn.click(
         fn = infer,
         inputs = [prompt],