update

Dockerfile

@@ -18,7 +18,7 @@ USER user
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
 
-ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/python3.8/site-packages/k2/lib/
+#ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/python3.8/site-packages/k2/lib/
 ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/user/.local/lib/python3.8/site-packages/k2/lib/
 
 

main.py

@@ -11,18 +11,6 @@ import platform
 import time
 import tempfile
 
-# os.system(
-#     "export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/python3.8/site-packages/k2/lib/"
-# )
-#
-# os.system(
-#     "cp -v /usr/local/lib/python3.8/site-packages/k2/lib/*.so /usr/local/lib/python3.8/site-packages/sherpa/lib/"
-# )
-#
-# os.system(
-#     "cp -v /home/user/.local/lib/python3.8/site-packages/k2/lib/*.so /home/user/.local/lib/python3.8/site-packages/sherpa/lib/"
-# )
-
 from project_settings import project_path, log_directory
 import log
 
@@ -113,15 +101,13 @@ def process(
 
     # load recognizer
     local_model_dir = pretrained_model_dir / "huggingface" / repo_id
-    nn_model_file = local_model_dir / m_dict["nn_model_file"]
-    tokens_file = local_model_dir / m_dict["tokens_file"]
 
     recognizer = nn_models.load_recognizer(
         repo_id=m_dict["repo_id"],
-        nn_model_file=nn_model_file.as_posix(),
-        tokens_file=tokens_file.as_posix(),
+        nn_model_file=m_dict["nn_model_file"],
+        tokens_file=m_dict["tokens_file"],
         sub_folder=m_dict["sub_folder"],
-        local_model_dir=local_model_dir.as_posix(),
+        local_model_dir=local_model_dir,
         loader=m_dict["loader"],
         decoding_method=decoding_method,
         num_active_paths=num_active_paths,

toolbox/k2_sherpa/nn_models.py

@@ -3,6 +3,7 @@
 from enum import Enum
 from functools import lru_cache
 import os
+from pathlib import Path
 
 import huggingface_hub
 import sherpa
@@ -107,7 +108,7 @@ def load_recognizer(repo_id: str,
                     nn_model_file: str,
                     tokens_file: str,
                     sub_folder: str,
-                    local_model_dir: str,
+                    local_model_dir: Path,
                     loader: str,
                     decoding_method: str = "greedy_search",
                     num_active_paths: int = 4,
@@ -118,9 +119,12 @@ def load_recognizer(repo_id: str,
             nn_model_file=nn_model_file,
             tokens_file=tokens_file,
             sub_folder=sub_folder,
-            local_model_dir=local_model_dir,
+            local_model_dir=local_model_dir.as_posix(),
         )
 
+    nn_model_file = (local_model_dir / nn_model_file).as_posix()
+    tokens_file = (local_model_dir / tokens_file).as_posix()
+
     if loader == "load_sherpa_offline_recognizer":
         recognizer = load_sherpa_offline_recognizer(
             nn_model_file=nn_model_file,