add app.py

app.py

@@ -29,7 +29,7 @@ def denoise(inputs):
             clean_chunk = MODEL(audio_chunk[None]).logits
         clean.append(clean_chunk.squeeze(0))
 
-    denoised = torch.concat(clean)[:, : audio.shape[-1]].squeeze().clamp(-1.0, 1.0)
+    denoised = torch.concat(clean).flatten()[: audio.shape[-1]].clamp(-1.0, 1.0)
     denoised = (denoised * 32767.0).numpy().astype("int16")
 
     print(f"Denoised shape: {denoised.shape}")