update

app.py

@@ -6,11 +6,15 @@ import os
 def separate_audio(file_path):
     output_dir = "output"
     model_dir = "audio_separator/pretrained_models/2stems"
-    # Run spleeter
+    # Set environment variable for Spleeter to use custom model directory
+    env = os.environ.copy()
+    env["SPLEETER_MODEL_PATH"] = model_dir
+
     result = subprocess.run(
-        ["spleeter", "separate", "-p", "spleeter:2stems", "-o", output_dir, "--model_dir", model_dir, file_path],
+        ["spleeter", "separate", "-p", "spleeter:2stems", "-o", output_dir, file_path],
         stdout=subprocess.PIPE,
         stderr=subprocess.PIPE,
+        env=env  # Pass modified environment
     )
 
     if result.returncode != 0: