vulnerability

Sleeping

leavoigt commited on Jan 7, 2024

Commit

a3014e0

1 Parent(s): 86c29fa

Update utils/target_classifier.py

Files changed (1) hide show

utils/target_classifier.py CHANGED Viewed

@@ -11,8 +11,8 @@ from transformers import pipeline
 ## Labels dictionary ###
 label_dict = {
-            '0':'NO',
-            '1':'YES',
             }
 def get_target_labels(preds):

 ## Labels dictionary ###
 label_dict = {
+            0:'NO',
+            1:'YES',
             }
 def get_target_labels(preds):