csukuangfj commited on
Commit
ae854f0
1 Parent(s): 4a82406

minor fixes

Browse files
Files changed (1) hide show
  1. model.py +0 -6
model.py CHANGED
@@ -273,12 +273,9 @@ def _get_aishell2_pretrained_model(
273
  def _get_whisper_model(
274
  repo_id: str, decoding_method: str, num_active_paths: int
275
  ) -> sherpa_onnx.OfflineRecognizer:
276
- print(f"get {repo_id}")
277
  name = repo_id.split("-")[1]
278
  assert name in ("tiny.en", "base.en", "small.en", "medium.en"), repo_id
279
- print(f"name: {name}")
280
  full_repo_id = "csukuangfj/sherpa-onnx-whisper-" + name
281
- print(f"full repo id: {full_repo_id}")
282
  encoder = _get_nn_model_filename(
283
  repo_id=full_repo_id,
284
  filename=f"{name}-encoder.int8.ort",
@@ -295,15 +292,12 @@ def _get_whisper_model(
295
  repo_id=full_repo_id, subfolder=".", filename=f"{name}-tokens.txt"
296
  )
297
 
298
- print("creating recognizer")
299
-
300
  recognizer = sherpa_onnx.OfflineRecognizer.from_whisper(
301
  encoder=encoder,
302
  decoder=decoder,
303
  tokens=tokens,
304
  num_threads=2,
305
  )
306
- print("created recognizer")
307
 
308
  return recognizer
309
 
 
273
  def _get_whisper_model(
274
  repo_id: str, decoding_method: str, num_active_paths: int
275
  ) -> sherpa_onnx.OfflineRecognizer:
 
276
  name = repo_id.split("-")[1]
277
  assert name in ("tiny.en", "base.en", "small.en", "medium.en"), repo_id
 
278
  full_repo_id = "csukuangfj/sherpa-onnx-whisper-" + name
 
279
  encoder = _get_nn_model_filename(
280
  repo_id=full_repo_id,
281
  filename=f"{name}-encoder.int8.ort",
 
292
  repo_id=full_repo_id, subfolder=".", filename=f"{name}-tokens.txt"
293
  )
294
 
 
 
295
  recognizer = sherpa_onnx.OfflineRecognizer.from_whisper(
296
  encoder=encoder,
297
  decoder=decoder,
298
  tokens=tokens,
299
  num_threads=2,
300
  )
 
301
 
302
  return recognizer
303