Add examples

app.py

@@ -7,6 +7,7 @@ import librosa
 import numpy as np
 import torch
 
+from gradio_examples import EXAMPLES
 from pipeline import build_audiosep
 
 CHECKPOINTS_DIR = Path("checkpoint")
@@ -59,11 +60,11 @@ with gr.Blocks(title="AudioSep") as demo:
     gr.Markdown(description)
     with gr.Row():
         with gr.Column():
-            input_audio = gr.Audio(type="filepath")
-            text = gr.Textbox()
+            input_audio = gr.Audio(label="Mixture", type="filepath")
+            text = gr.Textbox(label="Text Query")
         with gr.Column():
             with gr.Column():
-                output_audio = gr.Audio(scale=10)
+                output_audio = gr.Audio(label="Separation Result", scale=10)
                 button = gr.Button(
                     "Separate",
                     variant="primary",
@@ -75,4 +76,7 @@ with gr.Blocks(title="AudioSep") as demo:
                     fn=inference, inputs=[input_audio, text], outputs=[output_audio]
                 )
 
+    gr.Markdown("## Examples")
+    gr.Examples(examples=EXAMPLES, inputs=[input_audio, text])
+
 demo.queue().launch(share=True)

gradio_examples.py

@@ -0,0 +1,15 @@
+from pathlib import Path
+
+CURR_DIR = Path(__file__).resolve().parent
+
+EXAMPLES_DIR = CURR_DIR / "examples"
+
+EXAMPLES = [
+    [EXAMPLES_DIR / "acoustic_guitar.wav", "acoustic guitar"],
+    [EXAMPLES_DIR / "laughing.wav", "laughing"],
+    [
+        EXAMPLES_DIR / "ticktok_piano.wav",
+        "A ticktock sound playing at the same rhythm with piano.",
+    ],
+    [EXAMPLES_DIR / "water_drops.wav", "water drops"],
+]