csukuangfj commited on
Commit
c342af5
1 Parent(s): dcb78cd

Update sherpa-onnx

Browse files
Files changed (2) hide show
  1. model.py +6 -3
  2. 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 ("alphacep/vosk-model-ru", "alphacep/vosk-model-small-ru"), repo_id
 
 
 
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