Upload directory

README.md

@@ -35,6 +35,7 @@ from huggingface_hub import hf_hub_download
 import shutil
 import os
 import torch
+import sys
 
 
 # helpfer function to download huggingface repo and use model
@@ -55,8 +56,10 @@ def download(repo_id, path, HF_TOKEN=None):
 def load_model_from_local_path(path, HF_TOKEN=None):
     cwd = os.getcwd()
     os.chdir(path)
+    sys.path.insert(0, path)
     model = AutoModel.from_pretrained(path, trust_remote_code=True, token=HF_TOKEN)
     os.chdir(cwd)
+    sys.path.pop(0)
     return model