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