Fixed the model download in init_spacy

ssciwr · Oct 29, 2024 · 61732c9 · 61732c9
1 parent 2f13356
commit 61732c9
Showing 1 changed file with 12 additions and 15 deletions.
diff --git a/mailcom/parse.py b/mailcom/parse.py
@@ -67,21 +67,18 @@ def init_spacy(self, language: str, model="default"):
                 model, exclude=["morphologizer", "attribute_ruler", "lemmatizer", "ner"]
             )
         except OSError:
-            pass
-        try:
-            print(
-                "Could not find model in standard directory. Trying to download model from repo."  # noqa
-            )
-            # try downloading model
-            sp.cli.download(model)
-            self.nlp_spacy = sp.load(
-                model,
-                exclude=["morphologizer", "attribute_ruler", "lemmatizer", "ner"],
-            )
-        except SystemExit:
-            raise SystemExit("Could not download {} from repo".format(model))
-        except OSError:
-            raise OSError("Could not find {} in standard directory".format(model))
+            try:
+                print(
+                    "Could not find model in standard directory. Trying to download model from repo."  # noqa
+                )
+                # try downloading model
+                sp.cli.download(model)
+                self.nlp_spacy = sp.load(
+                    model,
+                    exclude=["morphologizer", "attribute_ruler", "lemmatizer", "ner"],
+                )
+            except SystemExit:
+                raise SystemExit("Could not download {} from repo".format(model))
 
     def init_transformers(
         self,