Mohidul Islam 4 anni fa
parent
commit
c47368f2bf
1 ha cambiato i file con 2 aggiunte e 2 eliminazioni
  1. 2 2
      src/models.py

+ 2 - 2
src/models.py

@@ -1,6 +1,6 @@
 import spacy
 
-MODEL_NAME = 'en_core_web_sm'
+MODEL_NAME = 'en_core_web_md'
 
 def get_model():
     if not spacy.util.is_package(MODEL_NAME):
@@ -8,4 +8,4 @@ def get_model():
         print(f"\033[92mSpacy model has downloaded!")
     ner_model = spacy.load(MODEL_NAME)
     print(f"Spacy model named {ner_model.meta.get('name')} has loaded!\033[0m")
-    return ner_model
+    return ner_model