cleaning up entity recognition output

app.py

@@ -9,10 +9,11 @@ def get_entities(article):
     
 def make_summary(article, max_length=130, min_length=30, do_sample=False):
     summ = summarizer(article, max_length, min_length, do_sample)[0]
-    ents = get_entities(article)[0]
-    return summ.get('summary_text'), type(ents), ents
+    ents = get_entities(article)
+    entslen = len(ents[0].keys())
+    return summ.get('summary_text'), entslen, ents
 
 
 
-iface = gr.Interface(fn=make_summary, inputs="text", outputs=['text', 'text', 'text'])
+iface = gr.Interface(fn=make_summary, inputs="text", outputs=['text', 'int', 'text'])
 iface.launch()