SpatialWebAgent commited on
Commit
3967cb4
·
verified ·
1 Parent(s): 3ae6a45

Update pages/2_Tagger.py

Browse files
Files changed (1) hide show
  1. pages/2_Tagger.py +3 -3
pages/2_Tagger.py CHANGED
@@ -13,7 +13,7 @@ import folium
13
  from utils import geoutil
14
  from disambiguation import disambiguate
15
  import spacy
16
- import en_core_web_md
17
  from geocoder import geo_level3
18
  from streamlit.components.v1 import html
19
  from spacy.tokens import Span, Doc
@@ -110,7 +110,7 @@ def set_selected_entities(doc, types):
110
 
111
  def extract_spatial_entities(model,text, types):
112
  # nlp = spacy.load(model)
113
- nlp = spacy.load('en_core_web_md')
114
  nlp.add_pipe("spatial_pipeline", after="ner")
115
 
116
  doc = nlp(text)
@@ -361,7 +361,7 @@ def call_backend(input_text):
361
  return {"error": f"⚠️ Connection Error: {str(e)}"}
362
 
363
  set_header()
364
- nlp = spacy.load("en_core_web_md")
365
  # doc = Doc(nlp.vocab).from_disk("saved_doc.spacy")
366
  doc = Doc(nlp.vocab).from_disk("/tmp/saved_doc.spacy")
367
 
 
13
  from utils import geoutil
14
  from disambiguation import disambiguate
15
  import spacy
16
+ # import en_core_web_md
17
  from geocoder import geo_level3
18
  from streamlit.components.v1 import html
19
  from spacy.tokens import Span, Doc
 
110
 
111
  def extract_spatial_entities(model,text, types):
112
  # nlp = spacy.load(model)
113
+ nlp = spacy.load('./en_core_web_md')
114
  nlp.add_pipe("spatial_pipeline", after="ner")
115
 
116
  doc = nlp(text)
 
361
  return {"error": f"⚠️ Connection Error: {str(e)}"}
362
 
363
  set_header()
364
+ nlp = spacy.load("./en_core_web_md")
365
  # doc = Doc(nlp.vocab).from_disk("saved_doc.spacy")
366
  doc = Doc(nlp.vocab).from_disk("/tmp/saved_doc.spacy")
367