seanpedrickcase commited on
Commit
773f8b8
1 Parent(s): 1963801

updated anon py with large model

Browse files
Files changed (1) hide show
  1. tools/anonymiser.py +3 -3
tools/anonymiser.py CHANGED
@@ -9,8 +9,8 @@ spacy.prefer_gpu()
9
 
10
  def spacy_model_installed(model_name):
11
  try:
12
- import en_core_web_sm
13
- en_core_web_sm.load()
14
  print("Successfully imported spaCy model")
15
  #nlp = spacy.load("en_core_web_sm")
16
  #print(nlp._path)
@@ -23,7 +23,7 @@ def spacy_model_installed(model_name):
23
 
24
  #if not is_model_installed(model_name):
25
  # os.system(f"python -m spacy download {model_name}")
26
- model_name = "en_core_web_sm"
27
  spacy_model_installed(model_name)
28
 
29
  #spacy.load(model_name)
 
9
 
10
  def spacy_model_installed(model_name):
11
  try:
12
+ import en_core_web_lg
13
+ en_core_web_lg.load()
14
  print("Successfully imported spaCy model")
15
  #nlp = spacy.load("en_core_web_sm")
16
  #print(nlp._path)
 
23
 
24
  #if not is_model_installed(model_name):
25
  # os.system(f"python -m spacy download {model_name}")
26
+ model_name = "en_core_web_lg"
27
  spacy_model_installed(model_name)
28
 
29
  #spacy.load(model_name)