add app.py

app.py

@@ -29,12 +29,7 @@ def denoise(inputs):
             clean_chunk = MODEL(audio_chunk[None]).logits
         clean.append(clean_chunk.squeeze(0))
 
-    denoised = (
-        torch.concat(clean)
-        .flatten()
-        .unsqueeze(0)[:, : audio.shape[-1]]
-        .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}")