nithinraok commited on
Commit
39684bb
·
verified ·
1 Parent(s): 4399430

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -94,6 +94,8 @@ def get_transcripts(audio_path, model):
94
  def pick_asr_model():
95
  model = 'nvidia/parakeet-tdt_ctc-1.1b'
96
  asr_model = ASRModel.from_pretrained(model).to(device)
 
 
97
  asr_model.eval()
98
  return asr_model
99
 
 
94
  def pick_asr_model():
95
  model = 'nvidia/parakeet-tdt_ctc-1.1b'
96
  asr_model = ASRModel.from_pretrained(model).to(device)
97
+ asr_model.cfg.decoding.strategy = "greedy_batch"
98
+ asr_model.change_decoding_strategy(asr_model.cfg.decoding)
99
  asr_model.eval()
100
  return asr_model
101