Update app.py

app.py

@@ -32,6 +32,8 @@ def split_process(audio, chosen_out_track):
         return "./out/mdx_extra_q/test/drums.wav"
     elif chosen_out_track == "other":
         return "./out/mdx_extra_q/test/other.wav"
+    elif chosen_out_track == "all-in":
+        return "test.wav"
     
 def load_model(version):
     print("Loading model", version)
@@ -97,7 +99,7 @@ with gr.Blocks(css=css) as demo:
             
         with gr.Column():
             uploaded_sound = gr.Audio(type="numpy", label="Input", source="upload")
-            chosen_track = gr.Radio(["vocals", "bass", "drums", "other"], label="Track", info="Which track from your audio do you want to mashup ?", value="vocals")
+            chosen_track = gr.Radio(["vocals", "bass", "drums", "other", "all-in"], label="Track", info="Which track from your audio do you want to mashup ?", value="vocals")
             load_sound_btn = gr.Button('Load your sound')
             #split_vocals = gr.Audio(type="filepath", label="Vocals")
             #split_bass = gr.Audio(type="filepath", label="Bass")
@@ -109,7 +111,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             model = gr.Radio(["melody", "medium", "small", "large"], label="Model", value="melody", interactive=True)
         with gr.Row():
-            duration = gr.Slider(minimum=1, maximum=30, value=10, step=1, label="Duration", interactive=True)
+            duration = gr.Slider(minimum=1, maximum=30, value=10, step=1, label="Generated Music Duration", interactive=True)
         with gr.Row():
             submit = gr.Button("Submit")
         #with gr.Row():