add app.py

app.py

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