Update app.py

app.py

@@ -9,7 +9,7 @@ model = AutoModelForTokenClassification.from_pretrained(model_name)
 model.to("cpu")  # Use "cuda" if you have GPU
 model.eval()
 
-# Extract token spans labeled as toponyms
+
 def get_toponym_entities(text):
     inputs = tokenizer(
         text,
@@ -30,10 +30,11 @@ def get_toponym_entities(text):
         if label_id != 0 and idx < len(offset_mapping):
             start, end = offset_mapping[idx].tolist()
             if end > start:
-                entities.append((start, end, "Toponym"))
+                entities.append({"start": start, "end": end, "entity": "Toponym"})
 
     return {"text": text, "entities": entities}
 
+
 # Launch Gradio app
 demo = gr.Interface(
     fn=get_toponym_entities,