update app

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import pipeline, AutoModelForTokenClassification, AutoTokenizer
+from transformers import pipeline, AutoTokenizer
 
 # Define the model name
 MODEL_NAME = "impresso-project/ner-stacked-bert-multilingual"
@@ -21,7 +21,11 @@ def prepare_entities_for_highlight(text, results):
     entities = []
     for category, entity_list in results.items():
         for entity in entity_list:
-            # Appending each entity's word, start and end for highlighting, including the entity label and score
+            # Debugging information to check character positions
+            print(
+                f"Entity: {entity['word']}, Start: {entity['start']}, End: {entity['end']}"
+            )
+            # Append entity with character indices
             entities.append(
                 {
                     "start": entity["start"],