update

examples/wenet/downaload_model.py

@@ -51,15 +51,28 @@ def main():
     pretrained_model_dir = Path(args.pretrained_model_dir)
     pretrained_model_dir.mkdir(exist_ok=True)
 
-    model_dir = pretrained_model_dir / "huggingface" / args.repo_id
-    model_dir.mkdir(parents=True, exist_ok=True)
+    repo_id: Path = Path(args.repo_id)
+    if len(repo_id.parts) == 1:
+        repo_name = repo_id.parts[-1]
+        repo_name = repo_name[:40]
+        folder = repo_name
+    elif len(repo_id.parts) == 2:
+        repo_supplier = repo_id.parts[-2]
+        repo_name = repo_id.parts[-1]
+        repo_name = repo_name[:40]
+        folder = "{}/{}".format(repo_supplier, repo_name)
+    else:
+        raise AssertionError("repo_id parts count invalid: {}".format(len(repo_id.parts)))
+
+    local_model_dir = pretrained_model_dir / "huggingface" / folder
+    local_model_dir.mkdir(parents=True, exist_ok=True)
 
     print("download model")
     model_filename = huggingface_hub.hf_hub_download(
         repo_id=args.repo_id,
         filename=args.model_filename,
         subfolder=args.model_sub_folder,
-        local_dir="E:/Users/tianx/HuggingSpaces/asr/pretrained_models/huggingface/luomingshuang/transducer_stateless2",
+        local_dir=local_model_dir.as_posix(),
     )
     print(model_filename)
     exit(0)

main.py

@@ -99,10 +99,23 @@ def process(
     if m_dict is None:
         raise AssertionError("repo_id invalid: {}".format(repo_id))
 
-    # load recognizer
-    local_model_dir = pretrained_model_dir / "huggingface" / repo_id
-    # local_model_dir.mkdir(parents=True, exist_ok=True)
+    # local_model_dir
+    repo_id: Path = Path(repo_id)
+    if len(repo_id.parts) == 1:
+        repo_name = repo_id.parts[-1]
+        repo_name = repo_name[:40]
+        folder = repo_name
+    elif len(repo_id.parts) == 2:
+        repo_supplier = repo_id.parts[-2]
+        repo_name = repo_id.parts[-1]
+        repo_name = repo_name[:40]
+        folder = "{}/{}".format(repo_supplier, repo_name)
+    else:
+        raise AssertionError("repo_id parts count invalid: {}".format(len(repo_id.parts)))
+
+    local_model_dir = pretrained_model_dir / "huggingface" / folder
 
+    # load recognizer
     recognizer = nn_models.load_recognizer(
         local_model_dir=local_model_dir,
         decoding_method=decoding_method,