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]
|