HoneyTian commited on
Commit
f92e324
1 Parent(s): f9690b9
Files changed (1) hide show
  1. main.py +3 -2
main.py CHANGED
@@ -150,10 +150,11 @@ def process(
150
 
151
  # load_punctuation_model
152
  if add_punctuation == "Yes":
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
  )
 
150
 
151
  # load_punctuation_model
152
  if add_punctuation == "Yes":
153
+ repo_id = "csukuangfj/sherpa-onnx-punct-ct-transformer-zh-en-vocab272727-2024-04-12"
154
+ local_model_dir = pretrained_model_dir / "huggingface" / md5_encrypt(repo_id)
155
  punctuation_model = nn_models.load_punctuation_model(
156
  local_model_dir=local_model_dir,
157
+ repo_id=repo_id,
158
  nn_model_file="model.onnx",
159
  nn_model_file_sub_folder=".",
160
  )