Commit
·
c342af5
1
Parent(s):
dcb78cd
Update sherpa-onnx
Browse files- model.py +6 -3
- requirements.txt +1 -1
model.py
CHANGED
@@ -272,11 +272,15 @@ def _get_aishell2_pretrained_model(
|
|
272 |
|
273 |
return recognizer
|
274 |
|
|
|
275 |
@lru_cache(maxsize=10)
|
276 |
def _get_russian_pre_trained_model(
|
277 |
repo_id: str, decoding_method: str, num_active_paths: int
|
278 |
) -> sherpa_onnx.OfflineRecognizer:
|
279 |
-
assert repo_id in (
|
|
|
|
|
|
|
280 |
|
281 |
if repo_id == "alphacep/vosk-model-ru":
|
282 |
model_dir = "am-onnx"
|
@@ -317,7 +321,6 @@ def _get_russian_pre_trained_model(
|
|
317 |
return recognizer
|
318 |
|
319 |
|
320 |
-
|
321 |
@lru_cache(maxsize=10)
|
322 |
def _get_whisper_model(
|
323 |
repo_id: str, decoding_method: str, num_active_paths: int
|
@@ -812,7 +815,7 @@ def _get_french_pre_trained_model(
|
|
812 |
|
813 |
tokens = _get_token_filename(repo_id=repo_id, subfolder=".")
|
814 |
|
815 |
-
recognizer = sherpa_onnx.OnlineRecognizer(
|
816 |
tokens=tokens,
|
817 |
encoder=encoder_model,
|
818 |
decoder=decoder_model,
|
|
|
272 |
|
273 |
return recognizer
|
274 |
|
275 |
+
|
276 |
@lru_cache(maxsize=10)
|
277 |
def _get_russian_pre_trained_model(
|
278 |
repo_id: str, decoding_method: str, num_active_paths: int
|
279 |
) -> sherpa_onnx.OfflineRecognizer:
|
280 |
+
assert repo_id in (
|
281 |
+
"alphacep/vosk-model-ru",
|
282 |
+
"alphacep/vosk-model-small-ru",
|
283 |
+
), repo_id
|
284 |
|
285 |
if repo_id == "alphacep/vosk-model-ru":
|
286 |
model_dir = "am-onnx"
|
|
|
321 |
return recognizer
|
322 |
|
323 |
|
|
|
324 |
@lru_cache(maxsize=10)
|
325 |
def _get_whisper_model(
|
326 |
repo_id: str, decoding_method: str, num_active_paths: int
|
|
|
815 |
|
816 |
tokens = _get_token_filename(repo_id=repo_id, subfolder=".")
|
817 |
|
818 |
+
recognizer = sherpa_onnx.OnlineRecognizer.from_transducer(
|
819 |
tokens=tokens,
|
820 |
encoder=encoder_model,
|
821 |
decoder=decoder_model,
|
requirements.txt
CHANGED
@@ -9,4 +9,4 @@ sentencepiece>=0.1.96
|
|
9 |
numpy
|
10 |
|
11 |
huggingface_hub
|
12 |
-
sherpa-onnx
|
|
|
9 |
numpy
|
10 |
|
11 |
huggingface_hub
|
12 |
+
sherpa-onnx>=1.7.0
|