Update app.py

app.py

@@ -67,10 +67,10 @@ def separate_audio_by_stem(audio_path, stem_count):
     
     elif stem_count == 4:
         
-        vocals_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/vocals.mp3"
-        drums_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/drums.mp3"
-        bass_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/bass.mp3"
-        other_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/other.mp3"
+        vocals_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/vocals.wav"
+        drums_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/drums.wav"
+        bass_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/bass.wav"
+        other_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/other.wav"
         
         print(f"{vocals_path=} \t exists: {os.path.exists(vocals_path)}")
         print(f"{drums_path=} \t exists: {os.path.exists(drums_path)}")
@@ -128,4 +128,4 @@ if uploaded_file is not None:
         st.write("Output Files:")
         for audio in separate_audios:
             st.write(audio['description'])
-            st.audio(audio['path'], format="audio/mp3", start_time=0)
+            st.audio(audio['path'], format="audio/m4a", start_time=0)