Spaces:
Sleeping
Sleeping
update
Browse files- main.py +1 -0
- toolbox/k2_sherpa/nn_models.py +2 -0
main.py
CHANGED
@@ -153,6 +153,7 @@ def process(
|
|
153 |
local_model_dir = pretrained_model_dir / "huggingface" / md5_encrypt("csukuangfj/sherpa-onnx-punct-ct-transformer-zh-en-vocab272727-2024-04-12")
|
154 |
punctuation_model = nn_models.load_punctuation_model(
|
155 |
local_model_dir=local_model_dir,
|
|
|
156 |
nn_model_file="model.onnx",
|
157 |
nn_model_file_sub_folder=".",
|
158 |
)
|
|
|
153 |
local_model_dir = pretrained_model_dir / "huggingface" / md5_encrypt("csukuangfj/sherpa-onnx-punct-ct-transformer-zh-en-vocab272727-2024-04-12")
|
154 |
punctuation_model = nn_models.load_punctuation_model(
|
155 |
local_model_dir=local_model_dir,
|
156 |
+
repo_id="csukuangfj/sherpa-onnx-punct-ct-transformer-zh-en-vocab272727-2024-04-12",
|
157 |
nn_model_file="model.onnx",
|
158 |
nn_model_file_sub_folder=".",
|
159 |
)
|
toolbox/k2_sherpa/nn_models.py
CHANGED
@@ -842,12 +842,14 @@ def load_recognizer(local_model_dir: Path,
|
|
842 |
|
843 |
@lru_cache(maxsize=15)
|
844 |
def load_punctuation_model(local_model_dir: Path,
|
|
|
845 |
nn_model_file: str,
|
846 |
nn_model_file_sub_folder: str,
|
847 |
):
|
848 |
if not local_model_dir.exists():
|
849 |
download_model(
|
850 |
local_model_dir=local_model_dir.as_posix(),
|
|
|
851 |
nn_model_file=nn_model_file,
|
852 |
nn_model_file_sub_folder=nn_model_file_sub_folder,
|
853 |
)
|
|
|
842 |
|
843 |
@lru_cache(maxsize=15)
|
844 |
def load_punctuation_model(local_model_dir: Path,
|
845 |
+
repo_id: str,
|
846 |
nn_model_file: str,
|
847 |
nn_model_file_sub_folder: str,
|
848 |
):
|
849 |
if not local_model_dir.exists():
|
850 |
download_model(
|
851 |
local_model_dir=local_model_dir.as_posix(),
|
852 |
+
repo_id=repo_id,
|
853 |
nn_model_file=nn_model_file,
|
854 |
nn_model_file_sub_folder=nn_model_file_sub_folder,
|
855 |
)
|