added confidence

generic_nel.py

@@ -111,8 +111,8 @@ class NelPipeline(Pipeline):
             num_beams=1,
             num_return_sequences=1,
             max_new_tokens=30,
-            return_dict_in_generate=True,
-            output_scores=True,
+            # return_dict_in_generate=True,
+            # output_scores=True,
         )
         # print(outputs)
         token_ids, scores = outputs.sequences, outputs.scores