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