Spaces:
Build error
Build error
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
|