wrice commited on
Commit
ab6d95c
·
1 Parent(s): eb75e68

add app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -26,6 +26,7 @@ def denoise(inputs):
26
  clean.append(clean_chunk.squeeze(0))
27
 
28
  denoised = torch.concat(clean)[:, :audio.shape[-1]].squeeze().clamp(-1.0, 1.0)
 
29
  return MODEL.config.sample_rate, denoised.numpy()
30
 
31
  iface = gr.Interface(fn=denoise, inputs="audio", outputs="audio")
 
26
  clean.append(clean_chunk.squeeze(0))
27
 
28
  denoised = torch.concat(clean)[:, :audio.shape[-1]].squeeze().clamp(-1.0, 1.0)
29
+ denoised = (denoised * 32767.0).numpy().astype("int16")
30
  return MODEL.config.sample_rate, denoised.numpy()
31
 
32
  iface = gr.Interface(fn=denoise, inputs="audio", outputs="audio")