add aggregation step

app.py

@@ -13,15 +13,17 @@ examples = [
 def ner(text):
     output_skills = token_skill_classifier(text)
     for result in output_skills:
-        if result.get("entity"):
-            tag = result["entity"]
-            result["entity"] = tag + "-skill"
+        if result.get("entity_group"):
+            tag = result["entity_group"]
+            result["entity"] = tag + "-Skill"
+            del result["entity_group"]
 
     output_knowledge = token_knowledge_classifier(text)
     for result in output_knowledge:
-        if result.get("entity"):
-            tag = result["entity"]
-            result["entity"] = tag + "-knowledge"
+        if result.get("entity_group"):
+            tag = result["entity_group"]
+            result["entity"] = tag + "-Knowledge"
+            del result["entity_group"]
 
     output = output_skills + output_knowledge
     return {"text": text, "entities": output}