Spaces:
Running
Running
Shunfeng Zheng
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -46,7 +46,7 @@ BASE_URL = ""
|
|
46 |
model = ""
|
47 |
types = ""
|
48 |
nlp = spacy.load("en_core_web_md")
|
49 |
-
|
50 |
gpe_selected = 'GPE'
|
51 |
loc_selected = 'loc'
|
52 |
rse_selected = 'rse'
|
@@ -184,6 +184,7 @@ def get_spatial_ent(doc):
|
|
184 |
|
185 |
doc.ents = new_ents
|
186 |
return doc
|
|
|
187 |
def extract_spatial_entities(text):
|
188 |
|
189 |
|
|
|
46 |
model = ""
|
47 |
types = ""
|
48 |
nlp = spacy.load("en_core_web_md")
|
49 |
+
|
50 |
gpe_selected = 'GPE'
|
51 |
loc_selected = 'loc'
|
52 |
rse_selected = 'rse'
|
|
|
184 |
|
185 |
doc.ents = new_ents
|
186 |
return doc
|
187 |
+
nlp.add_pipe("spatial_pipeline", after="ner")
|
188 |
def extract_spatial_entities(text):
|
189 |
|
190 |
|