Spaces:

wrice
/

audio_denoiser

Sleeping

wrice commited on Oct 12, 2023

Commit

244d52c

1 Parent(s): bf095ee

add app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,8 +34,10 @@ def denoise(inputs):
         .flatten()
         .unsqueeze(0)[:, : audio.shape[-1]]
         .clamp(-1.0, 1.0)
-        .numpy()
     )
     return MODEL.config.sample_rate, denoised

         .flatten()
         .unsqueeze(0)[:, : audio.shape[-1]]
         .clamp(-1.0, 1.0)
     )
+    denoised = (denoised * 32767.0).numpy().astype("int16")
+    print(f"Denoised shape: {denoised.shape}")
     return MODEL.config.sample_rate, denoised