[1bdb11]: / aitrika / utils / load_spacy_model.py

Download this file

18 lines (14 with data), 402 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
import spacy
def load_spacy_model():
"""
Load the spaCy model. Download it if not installed.
Returns:
spacy.lang.en.English: en_core_web_sm
"""
if not spacy.util.is_package("en_core_web_sm"):
spacy.cli.download("en_core_web_sm")
nlp = spacy.load("en_core_web_sm")
return nlp
else:
nlp = spacy.load("en_core_web_sm")
return nlp