Update app.py

app.py

@@ -21,7 +21,16 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
 
     yield None, stream_log("Starting separation process...")
     yield None, stream_log(f"Loading audio file: {audio_file}")
-    wav = load_track(audio_file, separator.samplerate, separator.audio_channels)
+    
+    # Load the audio file without specifying channels
+    wav = load_track(audio_file, separator.samplerate)
+    
+    # Check the number of channels and adjust if necessary
+    if wav.shape[0] == 1:
+        wav = wav.repeat(2, 1)  # If mono, duplicate to stereo
+    elif wav.shape[0] > 2:
+        wav = wav[:2]  # If more than 2 channels, keep only the first two
+
     ref = wav.mean(0)
     wav = (wav - ref.view(1, -1)).to(device)
     yield None, stream_log("Audio loaded successfully. Applying model...")