Update app.py

app.py

@@ -7,8 +7,9 @@ import subprocess
 from pydub import AudioSegment
 from typing import Tuple, Dict, List
 from demucs.apply import apply_model
-from demucs.separate import load_track, save_tracks
+from demucs.separate import load_track
 from demucs.pretrained import get_model
+from demucs.audio import save_audio  # Import save_audio from demucs.audio
 
 # check if cuda is available
 device: str = "cuda" if torch.cuda.is_available() else "cpu"
@@ -70,7 +71,7 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
     for stem, source in zip(separator.sources, sources):
         stem_path: str = os.path.join(output_dir, f"{stem}.wav")
         # demucs.api.save_audio(source, stem_path, samplerate=separator.samplerate)
-        save_tracks(source, stem_path, separator.samplerate)
+        save_audio(source, stem_path, separator.samplerate)  # Use save_audio
         stems[stem] = stem_path
 
     # mix the selected stems