Wataru commited on
Commit
6ce52bb
·
verified ·
1 Parent(s): 7a569cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def denoise_speech(audio):
31
 
32
  # If stereo, convert to mono
33
  if waveform.ndim > 1 and waveform.shape[0] > 1:
34
- waveform = torch.mean(waveform, dim=0)
35
 
36
  # Add a batch dimension
37
  waveform = waveform.view(1, -1)
 
31
 
32
  # If stereo, convert to mono
33
  if waveform.ndim > 1 and waveform.shape[0] > 1:
34
+ waveform = torch.mean(waveform, dim=1)
35
 
36
  # Add a batch dimension
37
  waveform = waveform.view(1, -1)