Add samples

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.wav filter=lfs diff=lfs merge=lfs -text

app.py

@@ -26,6 +26,8 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 MODEL_NAMES = ['qd_detr']
 FEATURES = ['clap']
 TOPK_MOMENT = 5
+sample_path = "sample_data/1a-ODBWMUAE.wav"
+sample_query = "Water cascades down from a waterfall."
 
 """
 Helper functions
@@ -129,15 +131,15 @@ def main():
 
                 with gr.Group():
                     gr.Markdown("## Audio and query")
-                    audio_input = gr.Audio(type='filepath')
+                    audio_input = gr.Audio(sample_path, type='filepath')
                     output = gr.Textbox(label='Audio processing progress')
-                    query_input = gr.Textbox(label='query')
+                    query_input = gr.Textbox(sample_query, label='query')
                     button = gr.Button("Retrieve moment", variant="primary")
 
             with gr.Column():
                 with gr.Group():
                     gr.Markdown("## Retrieved moments")
-
+                    gr.Markdown("Click on the moment button to listen to the trimmed audio.")
                     button_1 = gr.Button(value='moment 1', visible=False, elem_id='result_0')
                     button_2 = gr.Button(value='moment 2', visible=False, elem_id='result_1')
                     button_3 = gr.Button(value='moment 3', visible=False, elem_id='result_2')
@@ -157,6 +159,7 @@ def main():
                 button.click(predict, 
                             inputs=[query_input], 
                             outputs=[button_1, button_2, button_3, button_4, button_5])
+        demo.load(audio_upload, inputs=[audio_input], outputs=output)
 
     demo.launch()
 

sample_data/1a-ODBWMUAE.wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2a3175fc225f4e220c1e9604123ad1cc2f2d6d56330e65a4df01157b6672a0e2
+size 3201056