Spacy autocheck

- Downloads core-web-lg model automatically if not installed
hauntedcupoftea · Aug 17, 2024 · 807fb7e · 807fb7e
1 parent 618928b
commit 807fb7e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/django_vaani/connection/views.py b/django_vaani/connection/views.py
@@ -10,7 +10,14 @@
 
 
 import spacy
-nlp = spacy.load('en_core_web_lg')
+# Check if the 'en_core_web_lg' model is installed
+try:
+    nlp = spacy.load('en_core_web_lg')
+except OSError:
+    print("Model 'en_core_web_lg' not found. Installing now...")
+    spacy.cli.download('en_core_web_lg')
+    nlp = spacy.load('en_core_web_lg')
+    print("Model 'en_core_web_lg' installed successfully.")
 
 MAX_LENGTH = 256