Update app.py

app.py

@@ -39,9 +39,11 @@ if Run_Button == True:
     ner_pipeline, id2label = setModel(model_checkpoint, aggregation)
     output = ner_pipeline(input_text)
 
+    print(id2label)
+
     output_new = []
     for o in output:
-        o["entity_group"] = id2label[o["entity_group"].split("_")[-1]]
+        o["entity_group"] = id2label[str(o["entity_group"].split("_")[-1])]
         output_new.append(o)
 
     df = pd.DataFrame.from_dict(output_new)