Spaces:

SinaLab
/

wojood-api

Running

TymaaHammouda commited on 4 days ago

Commit

af2fbec

1 Parent(s): 7f85e18

Update local_dir name

Files changed (1) hide show

Nested/utils/helpers.py CHANGED Viewed

@@ -97,7 +97,7 @@ 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"])
-    checkpoint_path = hf_hub_download(repo_id="SinaLab/Nested", filename="checkpoints")
     tagger.load(checkpoint_path)
     return tagger, tag_vocab, train_config

     train_config.trainer_config["kwargs"]["loss"] = loss
     tagger = load_object(train_config.trainer_config["fn"], train_config.trainer_config["kwargs"])
+    checkpoint_path = hf_hub_download(repo_id="SinaLab/Nested", local_dir="checkpoints")
     tagger.load(checkpoint_path)
     return tagger, tag_vocab, train_config