Update app.py

app.py

@@ -38,8 +38,13 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
     ref = wav.mean(0)
     wav = (wav - ref.view(1, -1))
     yield None, stream_log("Audio loaded successfully. Applying model...")
-    sources = apply_model(separator, wav, device=device, progress=True)
-    sources = sources * ref.view(1, -1) + ref.view(1, -1)
+    
+    # Update the apply_model call with only the required arguments
+    sources = apply_model(separator, wav.to(device))
+    
+    # Remove the line that was causing the issue
+    # sources = sources * ref.view(1, -1) + ref.view(1, -1)
+    
     yield None, stream_log("Model applied. Processing stems...")
 
     output_dir: str = os.path.join("separated", model_name, os.path.splitext(os.path.basename(audio_file))[0])