Spaces:

TeresaK
/

cpv_test

Runtime error

leavoigt commited on Jan 4

Commit

65b4609

•

1 Parent(s): 7265775

Update utils/vulnerability_classifier.py

Files changed (1) hide show

utils/vulnerability_classifier.py CHANGED Viewed

@@ -127,7 +127,7 @@ def vulnerability_classification(haystack_doc:pd.DataFrame,
-        pred_labels = get_vulnerability_labelslabels(predictions)
         haystack_doc['Vulnerability Label'] = pred_labels
     #   placeholder = {}

+        pred_labels = get_vulnerability_labels(predictions)
         haystack_doc['Vulnerability Label'] = pred_labels
     #   placeholder = {}