wrice commited on
Commit
f9c578c
·
1 Parent(s): 4887a61

add app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -10,6 +10,9 @@ def denoise(inputs):
10
  audio = torch.from_numpy(audio)[None]
11
  audio = audio / 32768.0
12
 
 
 
 
13
  if sr != MODEL.config.sample_rate:
14
  audio = torchaudio.functional.resample(audio, sr, MODEL.config.sample_rate)
15
 
 
10
  audio = torch.from_numpy(audio)[None]
11
  audio = audio / 32768.0
12
 
13
+ if audio.shape[0] > 1:
14
+ audio = audio[0].unsqueeze(0)
15
+
16
  if sr != MODEL.config.sample_rate:
17
  audio = torchaudio.functional.resample(audio, sr, MODEL.config.sample_rate)
18