megatrump commited on
Commit
f1b8fcd
·
1 Parent(s): af35adf
Files changed (1) hide show
  1. api.py +2 -0
api.py CHANGED
@@ -208,6 +208,8 @@ async def process_audio(audio_data: bytes, language: str = "auto") -> str:
208
  # Convert to mono channel
209
  if waveform.shape[0] > 1:
210
  waveform = waveform.mean(dim=0)
 
 
211
  print(2, waveform.shape)
212
 
213
  # Convert to numpy array and normalize
 
208
  # Convert to mono channel
209
  if waveform.shape[0] > 1:
210
  waveform = waveform.mean(dim=0)
211
+ else:
212
+ waveform = np.squeeze(waveform)
213
  print(2, waveform.shape)
214
 
215
  # Convert to numpy array and normalize