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