handle multi-channel audio

app.py

@@ -44,7 +44,7 @@ def denoise(model_name, inputs):
     for i in tqdm(range(0, padded.shape[-1], chunk_size)):
         audio_chunk = padded[:, :, i : i + chunk_size]
         with torch.no_grad():
-            clean_chunk = model(audio_chunk).logits
+            clean_chunk = model(audio_chunk).audio
         clean.append(clean_chunk.squeeze(0))
 
     denoised = torch.concat(clean, 1)[:, : audio.shape[-1]].clamp(-1.0, 1.0)