Labels

app.py

@@ -58,6 +58,8 @@ def waveformer(audio, label_choices):
     return fs, output.squeeze(0).squeeze(0).to(torch.short).numpy()
 
 
-label_checkbox = gr.CheckboxGroup(choices=TARGETS)
-demo = gr.Interface(fn=waveformer, inputs=['audio', label_checkbox], outputs="audio")
+input_audio = gr.Audio(label="Input audio")
+label_checkbox = gr.CheckboxGroup(choices=TARGETS, label="Input target selection(s)")
+output_audio = gr.Audio(label="Output audio")
+demo = gr.Interface(fn=waveformer, inputs=[input_audio, label_checkbox], outputs=output_audio)
 demo.launch(show_error=True)