Update paths

Nested/utils/helpers.py

@@ -9,6 +9,7 @@ import json
 import random
 import numpy as np
 from argparse import Namespace
+from huggingface_hub import hf_hub_download
 
 
 def logging_config(log_file=None):
@@ -70,12 +71,14 @@ def load_checkpoint(model_path):
              vocab - arabicner.utils.data.Vocab - indexed tags
              train_config - argparse.Namespace - training configurations
     """
-    with open(os.path.join(model_path, "tag_vocab.pkl"), "rb") as fh:
+    with open("Nested/utils/tag_vocab.pkl", "rb") as fh:
         tag_vocab = pickle.load(fh)
 
     # Load train configurations from checkpoint
     train_config = Namespace()
-    with open(os.path.join(model_path, "args.json"), "r") as fh:
+    args_path = hf_hub_download(repo_id="SinaLab/Nested", filename="args.json")
+    
+    with open(args_path, "r") as fh:
         train_config.__dict__ = json.load(fh)
 
     # Initialize the loss function, not used for inference, but evaluation
@@ -94,7 +97,9 @@ def load_checkpoint(model_path):
     train_config.trainer_config["kwargs"]["loss"] = loss
 
     tagger = load_object(train_config.trainer_config["fn"], train_config.trainer_config["kwargs"])
-    tagger.load(os.path.join(model_path, "checkpoints"))
+    checkpoint_path = hf_hub_download(repo_id="SinaLab/Nested", filename="checkpoints/checkpoint_2.pt")
+
+    tagger.load(checkpoint_path)
     return tagger, tag_vocab, train_config