leavoigt commited on
Commit
2decf42
1 Parent(s): 65b4609

Update utils/vulnerability_classifier.py

Browse files
Files changed (1) hide show
  1. utils/vulnerability_classifier.py +3 -1
utils/vulnerability_classifier.py CHANGED
@@ -40,7 +40,9 @@ def get_vulnerability_labels(preds):
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
-
 
 
44
  return result
45
 
46
  @st.cache_resource
 
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
+
44
+ st.write("Here are the results")
45
+ st.write(results)
46
  return result
47
 
48
  @st.cache_resource