Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
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 |
|