leavoigt commited on
Commit
e052348
1 Parent(s): cdc5cd2

Update utils/vulnerability_classifier.py

Browse files
Files changed (1) hide show
  1. utils/vulnerability_classifier.py +1 -1
utils/vulnerability_classifier.py CHANGED
@@ -37,7 +37,7 @@ def get_vulnerability_labels(preds):
37
 
38
  # Get label names
39
  preds_list = preds.tolist()
40
-
41
  # Get the name of the group where the prediction is equal to "1"
42
  result = [label_dict[key] for key, value in enumerate(preds_list) if value == 1]
43
 
 
37
 
38
  # Get label names
39
  preds_list = preds.tolist()
40
+ st.write("preds_list")
41
  # Get the name of the group where the prediction is equal to "1"
42
  result = [label_dict[key] for key, value in enumerate(preds_list) if value == 1]
43