Update app.py

app.py

@@ -22,26 +22,14 @@ def separate_audio(audio_path):
     print(result)
 
     paths = []
-    if stem_count == 2:
-        paths.append(gradio_temp_path + f"/separated_audio/{audio_filename}/accompaniment.wav")
-        paths.append(gradio_temp_path + f"/separated_audio/{audio_filename}/vocals.wav")
-    elif stem_count == 4 or stem_count == 5:
-        paths.append(gradio_temp_path + f"/separated_audio/{audio_filename}/vocals.wav")
-        paths.append(gradio_temp_path + f"/separated_audio/{audio_filename}/drums.wav")
-        paths.append(gradio_temp_path + f"/separated_audio/{audio_filename}/bass.wav")
-        paths.append(gradio_temp_path + f"/separated_audio/{audio_filename}/other.wav")
-        if stem_count == 5:
-            paths.append(gradio_temp_path + f"/separated_audio/{audio_filename}/piano.wav")
-
-    # Return the audio file paths as a list of Audio components
+    paths.append(gradio_temp_path + f"/separated_audio/{audio_filename}/accompaniment.wav")
+    paths.append(gradio_temp_path + f"/separated_audio/{audio_filename}/vocals.wav")
     return [gr.Audio(path) for path in paths]
 
-# Define the Gradio interface
 iface = gr.Interface(
     fn=separate_audio,
-    inputs=[gr.Audio(type="filepath")],
+    inputs=[gr.Audio(type="filepath", label="Audio File")],
     outputs=[gr.Audio(label='Accompaniment'), gr.Audio(label='Vocals')]
 )
 
-# Launch the interface
 iface.launch()