Commit
·
873c8d4
1
Parent(s):
5cb3bb4
minor fixes
Browse files
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 |
|