Spaces:
Sleeping
Sleeping
Update utils/vulnerability_classifier.py
Browse files
utils/vulnerability_classifier.py
CHANGED
@@ -127,7 +127,7 @@ def vulnerability_classification(haystack_doc:pd.DataFrame,
|
|
127 |
|
128 |
|
129 |
|
130 |
-
pred_labels =
|
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 = {}
|