leavoigt commited on
Commit
65b4609
1 Parent(s): 7265775

Update utils/vulnerability_classifier.py

Browse files
Files changed (1) hide show
  1. utils/vulnerability_classifier.py +1 -1
utils/vulnerability_classifier.py CHANGED
@@ -127,7 +127,7 @@ def vulnerability_classification(haystack_doc:pd.DataFrame,
127
 
128
 
129
 
130
- pred_labels = get_vulnerability_labelslabels(predictions)
131
 
132
  haystack_doc['Vulnerability Label'] = pred_labels
133
  # placeholder = {}
 
127
 
128
 
129
 
130
+ pred_labels = get_vulnerability_labels(predictions)
131
 
132
  haystack_doc['Vulnerability Label'] = pred_labels
133
  # placeholder = {}