mfidabel commited on
Commit
5d8b688
verified
1 Parent(s): 46f32ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -26,8 +26,9 @@ def transcribe(audio):
26
  return "Espera a que la grabaci贸n termine de subirse al servidor !! Intentelo de nuevo en unos segundos"
27
 
28
  sr, y = audio
29
- y = y.astype(np.float16)
30
  y /= np.max(np.abs(y))
 
31
  return pipeline({"sampling_rate": sr, "raw": y}, generate_kwargs={"forced_decoder_ids": forced_decoder_ids}, max_new_tokens=255)["text"]
32
 
33
  examples = [
 
26
  return "Espera a que la grabaci贸n termine de subirse al servidor !! Intentelo de nuevo en unos segundos"
27
 
28
  sr, y = audio
29
+ y = y.astype(np.float32)
30
  y /= np.max(np.abs(y))
31
+ y = torch.from_numpy(y).to(torch.float16).to('cuda')
32
  return pipeline({"sampling_rate": sr, "raw": y}, generate_kwargs={"forced_decoder_ids": forced_decoder_ids}, max_new_tokens=255)["text"]
33
 
34
  examples = [