From e20c6d7c9ad5009478f706c2c91dc756d75972b2 Mon Sep 17 00:00:00 2001
From: David Berenstein <david.berenstein@pandoraintelligence.com>
Date: Tue, 10 May 2022 12:36:07 +0200
Subject: [PATCH] added AllenNLP spanbert to coref models

---
 crosslingual_coreference/CrossLingualPredictor.py | 5 +++++
 pyproject.toml                                    | 2 +-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/crosslingual_coreference/CrossLingualPredictor.py b/crosslingual_coreference/CrossLingualPredictor.py
index a77c6e0..6a0fc55 100644
--- 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
index be591f4..43b1b33 100644
--- 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 <david.berenstein@pandoraintelligence.com>"]
 license = "MIT"