Update app.py

app.py

@@ -84,7 +84,7 @@ def separate_audio_by_stem(audio_path, stem_count):
 st.write("Upload an audio file")
 
 uploaded_file = st.file_uploader("Choose a file", type=["wav","mp3"])
-selected_stem_count = st.radio("Select stem count", (2,3,5))
+selected_stem_count = st.radio("Select stem count", (2,4,5))
 
 if uploaded_file is not None:
 
@@ -94,11 +94,9 @@ if uploaded_file is not None:
         temp_file_path = temp_file.name
     
     # Process the uploaded audio file
-    separated_audio_paths = separate_audio(temp_file_path)
+    separated_audio_paths = separate_audio_by_stem(temp_file_path)
 
     # Display the output files for download
     st.write("Output Files:")
-    # st.audio(vocals_path, format="audio/wav", start_time=0)
-    # st.audio(accompaniment_path, format="audio/wav", start_time=0)
     for path in separated_audio_paths:
         st.audio(path, format="audio/wav", start_time=0)