handle multi-channel audio

app.py

@@ -26,8 +26,10 @@ def denoise(model_name, inputs):
     print(f"Audio shape: {audio.shape}")
     print(f"Sample rate: {sr}")
 
-    if audio.shape[1] > 1:
-        audio = audio.mean(1, keepdim=True)
+    if audio.shape[-1] > 1:
+        audio = audio.mean(-1, keepdim=True)
+
+    print(f"Audio shape: {audio.shape}")
 
     if sr != model.config.sample_rate:
         audio = torchaudio.functional.resample(audio, sr, model.config.sample_rate)