Update app.py

app.py

@@ -2,7 +2,7 @@ import time
 from pathlib import Path
 
 import gradio as gr
-import onnxruntime as rt
+# import onnxruntime as rt
 from audio_separator.separator import Separator
 from pydub import AudioSegment
 from pytube import YouTube
@@ -54,20 +54,21 @@ def audio_sep(youtube_url, audio_path, separation_model, separation_mode, progre
         separation_mode = None
     progress(0, desc="Starting...")
     separator = Separator(
-        audio_path,
-        model_name=separation_model,
-        use_cuda=True if rt.get_device() == "GPU" else False,
+        # audio_path,
+        # model_name=separation_model,
+        # use_cuda=True if rt.get_device() == "GPU" else False
         output_dir=str(out_folder),
         output_single_stem=separation_mode,
     )
-    for i in progress.tqdm(range(50)):
-        time.sleep(0.01)
+    separator.load_model(separation_model)
+    # for i in progress.tqdm(range(50)):
+    #     time.sleep(0.01)
 
-    results = [out_folder / p for p in separator.separate()]
+    results = [out_folder / p for p in separator.separate(audio_path)]
     print(results)
 
-    for i in progress.tqdm(range(50, 100)):
-        time.sleep(0.01)
+    # for i in progress.tqdm(range(50, 100)):
+    #     time.sleep(0.01)
 
     if separation_mode == "Instrument":
         instrument_stem_path = str(results[0])