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 = {}
|