Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -91,13 +91,14 @@ if text_input:
|
|
91 |
entities = [(ent.text, ent.label_, ent._.kb_qid, ent._.url_wikidata) for ent in doc_de.ents]
|
92 |
for entity in entities:
|
93 |
entity_string, entity_type, wikidata_id, wikidata_url = entity
|
94 |
-
|
95 |
-
|
96 |
-
|
97 |
-
|
98 |
-
|
99 |
-
|
100 |
-
|
|
|
101 |
doc_de = model(text_input)
|
102 |
# Map entities to a format similar to English output
|
103 |
entities = [(ent.text, ent.label_, ent._.kb_qid, ent._.url_wikidata) for ent in doc_de.ents]
|
|
|
91 |
entities = [(ent.text, ent.label_, ent._.kb_qid, ent._.url_wikidata) for ent in doc_de.ents]
|
92 |
for entity in entities:
|
93 |
entity_string, entity_type, wikidata_id, wikidata_url = entity
|
94 |
+
if wikidata_url:
|
95 |
+
# Ensure correct format for the German model
|
96 |
+
formatted_wikidata_url = wikidata_url.replace("https://", "http://")
|
97 |
+
entities_map[entity_string] = {"id": wikidata_id, "link": formatted_wikidata_url}
|
98 |
+
entity_data = get_entity_data(formatted_wikidata_url)
|
99 |
+
if entity_data is not None:
|
100 |
+
entities_data[entity_string] = entity_data
|
101 |
+
|
102 |
doc_de = model(text_input)
|
103 |
# Map entities to a format similar to English output
|
104 |
entities = [(ent.text, ent.label_, ent._.kb_qid, ent._.url_wikidata) for ent in doc_de.ents]
|