Update app.py

app.py

@@ -39,11 +39,11 @@ 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 with only the required arguments
-    sources = apply_model(separator, wav.to(device))
+    # Update the apply_model call to use it as an instance method
+    sources = separator.apply_model(wav.to(device))
     
-    # Remove the line that was causing the issue
-    # sources = sources * ref.view(1, -1) + ref.view(1, -1)
+    # Process the sources (no need to unpack as it's likely returning the correct format now)
+    sources = [source * ref.view(1, -1) + ref.view(1, -1) for source in sources]
     
     yield None, stream_log("Model applied. Processing stems...")