Spaces:
Running
Running
Update pages/2_Tagger.py
Browse files- 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 |
|