Update app.py

app.py

@@ -80,7 +80,7 @@ def process_ner(text: str, pipeline) -> dict:
     if current_entity is not None:
         entities.append(current_entity)
 
-    return {"entities": entities}
+    return {"text": text, "entities": entities}
 
 def process_classification(text: str, model1, model2, tokenizer1) -> Tuple[str, str, str]:
     inputs1 = tokenizer1(text, max_length=512, return_tensors='pt', truncation=True, padding=True)
@@ -95,8 +95,6 @@ def process_classification(text: str, model1, model2, tokenizer1) -> Tuple[str,
 
     return f"{round(prediction1, 1)}", f"{round(prediction2, 1)}", f"{round(score, 2)}"
 
-import plotly.graph_objects as go
-from typing import Tuple
 
 def generate_charts(ner_output_bin: dict, ner_output_ext: dict) -> Tuple[go.Figure, go.Figure, np.ndarray]:
     entities_bin = [entity['entity'] for entity in ner_output_bin['entities']]