leavoigt commited on
Commit
aec19dd
1 Parent(s): 3a33bff

Update utils/vulnerability_classifier.py

Browse files
Files changed (1) hide show
  1. utils/vulnerability_classifier.py +1 -1
utils/vulnerability_classifier.py CHANGED
@@ -42,7 +42,7 @@ def get_vulnerability_labels(preds):
42
  result = []
43
 
44
  for sublist in preds_list:
45
- names = [label_dict[key] for key, value in enumerate(sublist) if value == 1])
46
  result.append(names)
47
 
48
  st.write("Here are the results")
 
42
  result = []
43
 
44
  for sublist in preds_list:
45
+ names = [label_dict[key] for key, value in enumerate(sublist) if value == 1]
46
  result.append(names)
47
 
48
  st.write("Here are the results")