update for visualization

app.py

@@ -17,9 +17,13 @@ get_completion = pipeline(
 
 
 def ner(input):
-    output = get_completion(input)
-    print(output)
-    return {"text": input, "entities": output}
+    entities = get_completion(input)
+    print(entities)
+    for entity in entities:
+        entity["entity"] = entity["type"]
+        entity.pop("type")
+    print("After:", entities)
+    return {"text": input, "entities": entities}
 
 
 demo = gr.Interface(