Spaces:
Sleeping
Sleeping
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
|