Update app.py

app.py

@@ -37,7 +37,8 @@ def ner(input):
     idx = 0
     for i in output:
         result += text[idx:i["start"]]
-        result += text[i["start"]:i["end"]] + f"{id2label[i["entity_group"]]}"
+        ent = id2label[i["entity_group"]]
+        result += text[i["start"]:i["end"]] + f"{ent}"
         idx = i["end"]
     result+=text[idx:]
     print(result)