coref-spanbert-large-2021.03.10 folder

pipeline.py

@@ -4,13 +4,14 @@ from long_coref.coref.prediction import CorefPredictor
 from long_coref.coref.utils import ArchiveContent
 from allennlp.common.params import Params
 
+CHECKPOINT = "coref-spanbert-large-2021.03.10"
 
 class PreTrainedPipeline:
     def __init__(self, path=""):
         archive_content = ArchiveContent(
-            archive_dir=path,
-            weight_path=os.path.join(path, "weights.th"),
-            config=Params.from_file(os.path.join(path, "config.json")),
+            archive_dir=os.path.join(path, CHECKPOINT),
+            weight_path=os.path.join(path, CHECKPOINT, "weights.th"),
+            config=Params.from_file(os.path.join(path, CHECKPOINT, "config.json")),
         )
         self.predictor = CorefPredictor.from_extracted_archive(archive_content)