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