Spaces:
Build error
Build error
Commit
·
ae854f0
1
Parent(s):
4a82406
minor fixes
Browse files
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 |
|