wrice commited on
Commit
2094a60
1 Parent(s): 7ca9435

handle multi-channel audio

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -21,6 +21,8 @@ def denoise(model_name, inputs):
21
  model = WaveUNetModel.from_pretrained(model_name)
22
  sr, audio = inputs
23
  audio = torch.from_numpy(audio)[None]
 
 
24
  audio = audio / 32768.0
25
  audio = audio.permute(0, 2, 1)
26
 
 
21
  model = WaveUNetModel.from_pretrained(model_name)
22
  sr, audio = inputs
23
  audio = torch.from_numpy(audio)[None]
24
+ print(f"Audio shape: {audio.shape}")
25
+
26
  audio = audio / 32768.0
27
  audio = audio.permute(0, 2, 1)
28