소스 검색

chage model size

Mohidul Islam 4 년 전
부모
커밋
c47368f2bf
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  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