Update app.py

app.py

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