SpatialWebAgent commited on
Commit
2815bb6
·
verified ·
1 Parent(s): afa788d

Update pages/2_Tagger.py

Browse files
Files changed (1) hide show
  1. pages/2_Tagger.py +2 -1
pages/2_Tagger.py CHANGED
@@ -113,7 +113,8 @@ def set_selected_entities(doc, types):
113
 
114
  def extract_spatial_entities(model,text, types):
115
  # nlp = spacy.load(model)
116
- nlp = spacy.load('./en_core_web_md')
 
117
  nlp.add_pipe("spatial_pipeline", after="ner")
118
 
119
  doc = nlp(text)
 
113
 
114
  def extract_spatial_entities(model,text, types):
115
  # nlp = spacy.load(model)
116
+ model_path = os.path.join(os.getcwd(), "en_core_web_md")
117
+ nlp = spacy.load(model_path)
118
  nlp.add_pipe("spatial_pipeline", after="ner")
119
 
120
  doc = nlp(text)