updated dependencies to resolve pillow safety issue and allow for mor…

…e lenient requirements. Additionally, resolved small model download bug.
davidberenstein1957 · Jun 8, 2022 · b3e2e8b · b3e2e8b
1 parent ffa3afa
commit b3e2e8b
Show file tree

Hide file tree

Showing 4 changed files with 2,364 additions and 704 deletions.
diff --git a/crosslingual_coreference/CrossLingualPredictor.py b/crosslingual_coreference/CrossLingualPredictor.py
@@ -58,7 +58,10 @@ def download_model(self):
         """
         It downloads the model from the url provided and saves it in the current directory
         """
-        self.filename = self.model_url.replace("https://storage.googleapis.com/pandora-intelligence/", "")
+        if "https://storage.googleapis.com/pandora-intelligence/" in self.model_url:
+            self.filename = self.model_url.replace("https://storage.googleapis.com/pandora-intelligence/", "")
+        else:
+            self.filename = self.model_url.replace("https://storage.googleapis.com/allennlp-public-models/", "")
         path = pathlib.Path(self.filename)
         if path.is_file():
             pass

diff --git a/crosslingual_coreference/__init__.py b/crosslingual_coreference/__init__.py
@@ -1,3 +1,5 @@
+import warnings
+
 import nltk
 
 nltk.download("omw-1.4")
@@ -10,6 +12,8 @@
     CrossLingualPredictorSpacy as SpacyPredictor,
 )
 
+warnings.filterwarnings("ignore", category=UserWarning)
+warnings.filterwarnings("ignore", category=DeprecationWarning)
 __all__ = ["Predictor"]