fix examples

app.py

@@ -50,6 +50,9 @@ for name in wav_files:
 
 examples = [[model_list[0], w] for w in wav_files]
 
+logging.info(f"examples: {examples}")
+print(f"examples: {examples}")
+
 
 def build_html_output(s: str, style: str = "result_item_success"):
     return f"""
@@ -69,7 +72,7 @@ def process_url(model_name: str, url: str):
             return process(model_name, in_filename=f.name)
         except Exception as e:
             logging.info(str(e))
-            return "", build_html_output(str(e), "result_item_error")
+            return "", "", build_html_output(str(e), "result_item_error")
 
 
 def process_uploaded_file(model_name: str, in_filename: str):
@@ -85,7 +88,7 @@ def process_uploaded_file(model_name: str, in_filename: str):
         return process(model_name, in_filename=in_filename)
     except Exception as e:
         logging.info(str(e))
-        return "", build_html_output(str(e), "result_item_error")
+        return "", "", build_html_output(str(e), "result_item_error")
 
 
 def process_microphone(model_name: str, in_filename: str):
@@ -102,7 +105,7 @@ def process_microphone(model_name: str, in_filename: str):
         return process(model_name, in_filename=in_filename)
     except Exception as e:
         logging.info(str(e))
-        return "", build_html_output(str(e), "result_item_error")
+        return "", "", build_html_output(str(e), "result_item_error")
 
 
 def process(model_name, in_filename: str):
@@ -110,7 +113,6 @@ def process(model_name, in_filename: str):
     logging.info(f"in_filename: {in_filename}")
 
     samples, sample_rate = load_audio(in_filename)
-    samples = np.transpose(samples)
     samples = np.ascontiguousarray(samples)
 
     duration = samples.shape[1] / sample_rate  # in seconds
@@ -185,7 +187,7 @@ with demo:
     gr.Markdown(title)
 
     model_dropdown = gr.Dropdown(
-        choices=model_list[0],
+        choices=model_list,
         label="Select a model",
         value=model_list[0],
     )
@@ -246,9 +248,18 @@ with demo:
 
             gr.Examples(
                 examples=[
-                    "https://huggingface.co/csukuangfj/spleeter-torch/resolve/main/test_wavs/yesterday-once-more-Carpenters.mp3",
-                    "https://huggingface.co/csukuangfj/spleeter-torch/resolve/main/test_wavs/das-beste-Silbermond.mp3",
-                    "https://huggingface.co/csukuangfj/spleeter-torch/resolve/main/test_wavs/hotel-in-california.mp3",
+                    [
+                        model_list[0],
+                        "https://huggingface.co/csukuangfj/spleeter-torch/resolve/main/test_wavs/yesterday-once-more-Carpenters.mp3",
+                    ],
+                    [
+                        model_list[0],
+                        "https://huggingface.co/csukuangfj/spleeter-torch/resolve/main/test_wavs/das-beste-Silbermond.mp3",
+                    ],
+                    [
+                        model_list[0],
+                        "https://huggingface.co/csukuangfj/spleeter-torch/resolve/main/test_wavs/hotel-in-california.mp3",
+                    ],
                 ],
                 inputs=[model_dropdown, url_textbox],
                 outputs=[url_vocals, url_non_vocals, recorded_html_info],

separate.py

@@ -46,7 +46,7 @@ def load_audio(filename):
 def get_file(
     repo_id: str,
     filename: str,
-    subfolder: str = "2stems",
+    subfolder: str = ".",
 ) -> str:
     nn_model_filename = hf_hub_download(
         repo_id=repo_id,
@@ -95,7 +95,7 @@ def load_spleeter_model(name: str):
     elif "int8" in name:
         suffix = "int8.onnx"
     else:
-        suffix = ".onnx"
+        suffix = "onnx"
 
     vocals = get_file(repo_id=f"csukuangfj/{name}", filename=f"vocals.{suffix}")
     accompaniment = get_file(