Update function

app.py

@@ -2,10 +2,27 @@ import spacy
 import gradio as gr
 
 
+def make_ent_dict(list):
+    ent_dict = {
+    'ORG': '',
+    'PERSON': '',
+    'GPE': '',
+    'DATE': '',
+    'TIME': '',
+    'MONEY': '',
+    'PERCENT': '',
+}
+    for item in list:
+        if item[0] in ent_dict.keys():
+            ent_dict[item[0]] = item[1]
+    return ent_dict
+
+
 def get_named_entity(text):
     nlp = spacy.load("en_core_web_sm")
     doc = nlp(text)
-    return [(ent.text, ent.label_) for ent in doc.ents]
+    ent_list = [(ent.label_, ent.text) for ent in doc.ents]
+    return make_ent_dict(ent_list)
 
 
 iface = gr.Interface(fn=get_named_entity, inputs="text", outputs="text")