added AllenNLP spanbert to coref models

davidberenstein1957 · May 10, 2022 · e20c6d7 · e20c6d7
1 parent 7463efb
commit e20c6d7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/crosslingual_coreference/CrossLingualPredictor.py b/crosslingual_coreference/CrossLingualPredictor.py
@@ -27,6 +27,11 @@
         "f1_score_ontonotes": 74,
         "file_extension": ".tar.gz",
     },
+    "spanbert": {
+        "url": "https://storage.googleapis.com/allennlp-public-models/coref-spanbert-large-2021.03.10.tar.gz",
+        "f1_score_ontonotes": 83,
+        "file_extension": ".tar.gz",
+    },
 }
 
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "crosslingual-coreference"
-version = "0.2.3"
+version = "0.2.4"
 description = "A multi-lingual approach to AllenNLP CoReference Resolution, along with a wrapper for spaCy."
 authors = ["David Berenstein <[email protected]>"]
 license = "MIT"