csukuangfj commited on
Commit
873c8d4
1 Parent(s): 5cb3bb4

minor fixes

Browse files
Files changed (1) hide show
  1. model.py +5 -0
model.py CHANGED
@@ -276,7 +276,9 @@ def _get_whisper_model(
276
  print(f"get {repo_id}")
277
  assert repo_id in ("tiny.en", "base.en", "small.en", "medium.en"), repo_id
278
  name = repo_id.split("-")[1]
 
279
  full_repo_id = "csukuangfj/sherpa-onnx-whisper-" + name
 
280
  encoder = _get_nn_model_filename(
281
  repo_id=full_repo_id,
282
  filename=f"{name}-encoder.int8.ort",
@@ -293,12 +295,15 @@ def _get_whisper_model(
293
  repo_id=full_repo_id, subfolder=".", filename=f"{name}-tokens.txt"
294
  )
295
 
 
 
296
  recognizer = sherpa_onnx.OfflineRecognizer.from_whisper(
297
  encoder=encoder,
298
  decoder=decoder,
299
  tokens=tokens,
300
  num_threads=2,
301
  )
 
302
 
303
  return recognizer
304
 
 
276
  print(f"get {repo_id}")
277
  assert repo_id in ("tiny.en", "base.en", "small.en", "medium.en"), repo_id
278
  name = repo_id.split("-")[1]
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
  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