Update utils/vulnerability_classifier.py
Browse files
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
|