Update app.py

app.py

@@ -39,10 +39,10 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
     wav = (wav - ref.view(1, -1))
     yield None, stream_log("Audio loaded successfully. Applying model...")
     
-    # Update the apply_model call to use it as an instance method
-    sources = separator.apply_model(wav.to(device))
+    # Use apply_model as a standalone function
+    sources = apply_model(separator, wav.to(device), device=device)
     
-    # Process the sources (no need to unpack as it's likely returning the correct format now)
+    # Process the sources
     sources = [source * ref.view(1, -1) + ref.view(1, -1) for source in sources]
     
     yield None, stream_log("Model applied. Processing stems...")