Update app.py

app.py

@@ -1,4 +1,5 @@
 from flask import Flask, request, send_file
+import soundfile as sf
 import os
 import requests
 import subprocess
@@ -26,8 +27,21 @@ def audio_separation():
         	f.write(response.content)
 
         # 执行音频分离操作
-        separate.Predictor(output=Path("/tmp"), model_path=Path("./models/MDX_Net_Models/UVR-MDX-NET-Inst_HQ_3.onnx")).process_file(Path(mp3_filename))
-        subprocess.run(['python', 'separate.py', '/tmp/' + mp3_filename, '-o', '/tmp', '-m', './models/MDX_Net_Models/UVR-MDX-NET-Inst_HQ_3.onnx'])
+        audio_worker = separate.Predictor(args={
+    											"files": [f"/tmp/{mp3_filename}"],
+    											"output": Path("/tmp"),
+    											"model_path": Path("./models/MDX_Net_Models/UVR-MDX-NET-Inst_HQ_3.onnx"),
+    											"denoise": False,
+    											"margin": 44100,
+    											"chunks": 15,
+    											"n_fft": 6144,
+    											"dim_t": 8,
+    											"dim_f": 2048
+												})
+        vocals, no_vocals, sampling_rate = audio_worker.predict("/tmp/" + mp3_filename)
+        sf.write(os.path.join("/tmp", mp3_filename + "_no_vocals.wav"), no_vocals, sampling_rate)
+        sf.write(os.path.join("/tmp", mp3_filename + "_vocals.wav"), vocals, sampling_rate)
+        # subprocess.run(['python', 'separate.py', '/tmp/' + mp3_filename, '-o', '/tmp', '-m', './models/MDX_Net_Models/UVR-MDX-NET-Inst_HQ_3.onnx'])
 
     	# 生成分离后的文件名
         vocals_filename = f"{os.path.splitext(mp3_filename)[0]}_vocals.wav"