diff --git a/asr/asr_base.py b/asr/asr_base.py index d59acea..0b69cd9 100644 --- a/asr/asr_base.py +++ b/asr/asr_base.py @@ -7,6 +7,7 @@ from .asr_observer import AsrObserver class AsrBase: def __init__(self): + self._hot_words_file = '' self._sample_rate = 32000 self._samples_per_read = 100 self._observers = [] diff --git a/asr/sherpa_ncnn_asr.py b/asr/sherpa_ncnn_asr.py index 6306b93..9dd2e67 100644 --- a/asr/sherpa_ncnn_asr.py +++ b/asr/sherpa_ncnn_asr.py @@ -37,12 +37,12 @@ class SherpaNcnnAsr(AsrBase): joiner_param=base_path + '/joiner_jit_trace-pnnx.ncnn.param', joiner_bin=base_path + '/joiner_jit_trace-pnnx.ncnn.bin', num_threads=4, - decoding_method="modified_beam_search", + decoding_method='modified_beam_search', enable_endpoint_detection=True, rule1_min_trailing_silence=2.4, rule2_min_trailing_silence=1.2, rule3_min_utterance_length=300, - hotwords_file="", + hotwords_file=self._hot_words_file, hotwords_score=1.5, ) return recognizer