megatrump commited on
Commit
7520e98
·
1 Parent(s): 444b857

try to fix shape

Browse files
Files changed (1) hide show
  1. api.py +3 -2
api.py CHANGED
@@ -215,14 +215,15 @@ async def process_audio(audio_data: bytes, language: str = "auto") -> str:
215
  resampler = torchaudio.transforms.Resample(sample_rate, 16000)
216
  input_wav = resampler(torch.from_numpy(input_wav)[None, :])[0, :].numpy()
217
 
218
- print(input_wav.shape)
219
 
220
  target_length = 90 * 16000
221
  current_length = input_wav.shape[1]
222
  if current_length < target_length:
223
  padding_length = target_length - current_length
224
  padding = np.zeros((1, padding_length), dtype=np.float32)
225
- input_wav = np.concatenate((input_wav, padding))
 
 
226
 
227
  # Model inference
228
  text = model.generate(
 
215
  resampler = torchaudio.transforms.Resample(sample_rate, 16000)
216
  input_wav = resampler(torch.from_numpy(input_wav)[None, :])[0, :].numpy()
217
 
 
218
 
219
  target_length = 90 * 16000
220
  current_length = input_wav.shape[1]
221
  if current_length < target_length:
222
  padding_length = target_length - current_length
223
  padding = np.zeros((1, padding_length), dtype=np.float32)
224
+ print(input_wav.shape)
225
+ print(padding.shape)
226
+ input_wav = np.concatenate((input_wav, padding), axis=0)
227
 
228
  # Model inference
229
  text = model.generate(