fix: remove st_local_audio, convert orig audio to mp3 in inference script

app/pages/Separate.py

@@ -13,7 +13,6 @@ from helpers import (
     load_audio_segment,
     load_list_of_songs,
     plot_audio,
-    st_local_audio,
     url_is_valid,
     file_size_is_valid,
     delete_old_files,
@@ -34,7 +33,10 @@ label_sources = {
 }
 
 separation_mode_to_model = {
-    "Vocals & Instrumental (Faster)": ("vocal_remover", ["vocals.mp3", "no_vocals.mp3"]),
+    "Vocals & Instrumental (Low Quality, Faster)": (
+        "vocal_remover",
+        ["vocals.mp3", "no_vocals.mp3"],
+    ),
     "Vocals & Instrumental (High Quality, Slower)": ("htdemucs", ["vocals.mp3", "no_vocals.mp3"]),
     "Vocals, Drums, Bass & Other (Slower)": (
         "htdemucs",
@@ -85,7 +87,7 @@ def show_results(model_name: str, dir_name_output: str, file_sources: List):
                     use_column_width="always",
                 )
             with cols[1]:
-                st_local_audio(pathname, key=f"output_{file}_{dir_name_output}")
+                st.audio(str(pathname))
     log.info(f"Displaying results for {dir_name_output} - {model_name}")
 
 
@@ -195,7 +197,7 @@ def body():
         separation_mode = st.selectbox(
             "Choose the separation mode",
             [
-                "Vocals & Instrumental (Faster)",
+                "Vocals & Instrumental (Low Quality, Faster)",
                 "Vocals & Instrumental (High Quality, Slower)",
                 "Vocals, Drums, Bass & Other (Slower)",
                 "Vocal, Drums, Bass, Guitar, Piano & Other (Slowest)",
@@ -203,7 +205,7 @@ def body():
             on_change=reset_execution(),
             key="separation_mode",
         )
-        if separation_mode == "Vocals & Instrumental (Faster)":
+        if separation_mode == "Vocals & Instrumental (Low Quality, Faster)":
             max_duration = 30
         else:
             max_duration = 15

scripts/inference.py

@@ -1,7 +1,9 @@
 import argparse
 from pathlib import Path
-
 import warnings
+
+from pydub import AudioSegment
+
 from app.service.vocal_remover.runner import load_model, separate
 from app.service.demucs_runner import separator
 
@@ -10,6 +12,11 @@ warnings.simplefilter("ignore", FutureWarning)
 warnings.filterwarnings("ignore", module="streamlit")
 
 
+def convert_to_mp3(input_file, output_file):
+    audio = AudioSegment.from_file(input_file)
+    audio.export(output_file, format="mp3")
+
+
 def main():
     p = argparse.ArgumentParser()
     p.add_argument("--gpu", "-g", type=int, default=-1)
@@ -53,6 +60,7 @@ def main():
             output_dir=args.output_dir,
             only_no_vocals=True,
         )
+    convert_to_mp3(input_file, input_file)
 
 
 if __name__ == "__main__":