cyberandy commited on
Commit
890d925
1 Parent(s): b126447

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
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
- # Ensure correct format for the German model
95
- formatted_wikidata_url = wikidata_url.replace("https://", "http://")
96
- entities_map[entity_string] = {"id": wikidata_id, "link": formatted_wikidata_url}
97
- entity_data = get_entity_data(formatted_wikidata_url)
98
- if entity_data is not None:
99
- entities_data[entity_string] = entity_data
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]