Update app.py

app.py

@@ -23,7 +23,7 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
     yield None, stream_log(f"Loading audio file: {audio_file}")
     
     # Load the audio file with the correct samplerate
-    wav = load_track(audio_file, separator.samplerate)
+    wav, sr = load_track(audio_file)
     
     # Check the number of channels and adjust if necessary
     if wav.shape[0] == 1:
@@ -34,7 +34,7 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
     ref = wav.mean(0)
     wav = (wav - ref.view(1, -1)).to(device)
     yield None, stream_log("Audio loaded successfully. Applying model...")
-    sources = apply_model(separator, wav, device=device, progress=True)
+    sources = apply_model(separator, wav, device=device, progress=True, samplerate=sr)
     sources = sources * ref.view(1, -1) + ref.view(1, -1)
     yield None, stream_log("Model applied. Processing stems...")