From 879bd5c82536fbdc747df05383fc917f5a72c2e0 Mon Sep 17 00:00:00 2001 From: brige Date: Thu, 10 Oct 2024 11:47:52 +0800 Subject: [PATCH] modify code style --- asr/asr_base.py | 1 + asr/sherpa_ncnn_asr.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) 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