Update unknown_intents.py

GT4SD · Oct 28, 2022 · f3039db · f3039db
1 parent a60f902
commit f3039db
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/zberta/intent_discovery/unknown_intents.py b/src/zberta/intent_discovery/unknown_intents.py
@@ -8,9 +8,12 @@
 from nltk.corpus import wordnet
 
 default_spacy = "en_core_web_trf"
-os.system("python3 -m spacy download en_core_web_trf --no-deps")
 os.system("python -m spacy download en_core_web_trf --no-deps")
-nlp_engine = spacy.load("en_core_web_trf")
+try:
+    nlp_engine = spacy.load("en_core_web_trf")
+except:
+    os.system("python3 -m spacy download en_core_web_trf --no-deps")
+    nlp_engine = spacy.load("en_core_web_trf")
 lemmatizer = WordNetLemmatizer()