Update app.py

app.py

@@ -26,7 +26,12 @@ BATCHED_DURATION = 15
 INTERRUPTING = False
 # We have to wrap subprocess call to clean a bit the log when using gr.make_waveform
 _old_call = sp.call
-
+files = [
+    "./out/mdx_extra_q/test/vocals.wav",
+    "./out/mdx_extra_q/test/bass.wav",
+    "./out/mdx_extra_q/test/drums.wav",
+    "./out/mdx_extra_q/test/other.wav"
+]
 
 def _call_nostderr(*args, **kwargs):
     # Avoid ffmpeg vomitting on the logs.
@@ -233,12 +238,8 @@ def ui_full(launch_kwargs):
                     temperature = gr.Number(label="Temperature", value=1.0, interactive=True)
                     cfg_coef = gr.Number(label="Classifier Free Guidance", value=3.0, interactive=True)
             with gr.Column():
-                output = [
-    gr.Audio(audio_file1, label="Generated Music 1"),
-    gr.Audio(audio_file2, label="Generated Music 2"),
-    gr.Audio(audio_file3, label="Generated Music 3"),
-    gr.Audio(audio_file4, label="Generated Music 4")
-]
+                output = [gr.Audio(file, label=f"Generated Music {i+1}") for i, file in enumerate(files)]
+
         submit.click(predict_full,
                      inputs=[model, text, melody, duration, topk, topp, temperature, cfg_coef],
                      outputs=[output])
@@ -332,7 +333,7 @@ def ui_batched(launch_kwargs):
                 with gr.Row():
                     submit = gr.Button("Generate")
             with gr.Column():
-                output = gr.Audio(label="Generated Music")
+                output = [gr.Audio(file, label=f"Generated Music {i+1}") for i, file in enumerate(files)]
         submit.click(predict_batched, inputs=[text, melody],
                      outputs=[output], batch=True, max_batch_size=MAX_BATCH_SIZE)
         radio.change(toggle_audio_src, radio, [melody], queue=False, show_progress=False)