LuisG07 commited on
Commit
941714c
1 Parent(s): 34e7595

fix batch decoding

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,9 +34,9 @@ def predict_and_ctc_lm_decode(input_file, model_name):
34
  input_values = processor(speech, return_tensors="pt", sampling_rate=16000).input_values
35
  logits = model(input_values).logits.cpu().detach().numpy()[0]
36
 
37
- pred = processor.batch_decode(logits).text
38
 
39
- transcribed_text = fix_transcription_casing(pred[0].lower())
40
 
41
  return transcribed_text
42
 
 
34
  input_values = processor(speech, return_tensors="pt", sampling_rate=16000).input_values
35
  logits = model(input_values).logits.cpu().detach().numpy()[0]
36
 
37
+ pred = processor.decode(logits).text
38
 
39
+ transcribed_text = fix_transcription_casing(pred.lower())
40
 
41
  return transcribed_text
42