outputs mp3, name changes

README.md

@@ -28,7 +28,7 @@ A web-based tool for removing vocals from audio files using deep learning.
 
 ## Overview
 
-The Vocal Remover is a user-friendly web application that leverages deep learning models to remove vocals from audio files. It provides an easy and interactive way for users to upload their audio files and process them to obtain vocals-free versions.
+The Vocal Isolator is a user-friendly web application that leverages deep learning models to isolate vocals from audio files. It provides an easy and interactive way for users to upload their audio files and process them to obtain vocals-free versions.
 
 ## Features
 

app.py

@@ -3,6 +3,7 @@ import os
 
 # Third-Party
 import streamlit as st
+import librosa
 
 # Local
 from src.models.MDX_net.kimvocal import KimVocal
@@ -19,10 +20,10 @@ OUTPUT_FOLDER = "./datasets/output"
 def main():
     # Set page configuration and theming
     st.set_page_config(
-        page_title="Jarod's Vocal Remover",
+        page_title="Sing For Me",
         page_icon="🎵",
     )
-    st.title("Vocal Remover")
+    st.title("Vocal Isolator")
 
     # Upload WAV file
     uploaded_file = st.file_uploader(
@@ -32,7 +33,6 @@ def main():
     )
 
     if uploaded_file is not None:
-        uploaded_file_ext = uploaded_file.name.lower().split(".")[-1]
         # Process the uploaded audio
         st.subheader("Audio Processing")
         st.write("Processing the uploaded audio file...")
@@ -74,7 +74,8 @@ def main():
 
         # Display processed audio
         st.subheader("Processed Audio")
-        st.audio(vocals_array, format="audio/wav", sample_rate=samplerate)
+        # TODO: Is it encoding it wrong? Maybe fix it later.
+        st.audio(data=vocals_array, format="audio/mpeg", sample_rate=samplerate)
 
 
 if __name__ == "__main__":

requirements.txt

@@ -1,5 +1,6 @@
 torch~=2.0.1
 onnxruntime~=1.15.1
+onnxruntime-gpu~=1.15.1
 librosa~=0.10.0.post2
 soundfile~=0.12.1
 numpy~=1.24.4