Spaces:

intelli-zen
/

asr

Sleeping

HoneyTian commited on May 17, 2024

Commit

e67d20d

1 Parent(s): f92e324

update

Files changed (1) hide show

main.py CHANGED Viewed

@@ -150,11 +150,11 @@ def process(
     # load_punctuation_model
     if add_punctuation == "Yes":
-        repo_id = "csukuangfj/sherpa-onnx-punct-ct-transformer-zh-en-vocab272727-2024-04-12"
-        local_model_dir = pretrained_model_dir / "huggingface" / md5_encrypt(repo_id)
         punctuation_model = nn_models.load_punctuation_model(
             local_model_dir=local_model_dir,
-            repo_id=repo_id,
             nn_model_file="model.onnx",
             nn_model_file_sub_folder=".",
         )

     # load_punctuation_model
     if add_punctuation == "Yes":
+        punctuation_repo_id = "csukuangfj/sherpa-onnx-punct-ct-transformer-zh-en-vocab272727-2024-04-12"
+        local_model_dir = pretrained_model_dir / "huggingface" / md5_encrypt(punctuation_repo_id)
         punctuation_model = nn_models.load_punctuation_model(
             local_model_dir=local_model_dir,
+            repo_id=punctuation_repo_id,
             nn_model_file="model.onnx",
             nn_model_file_sub_folder=".",
         )