Update utils/vulnerability_classifier.py

utils/vulnerability_classifier.py

@@ -40,7 +40,10 @@ def get_vulnerability_labels(preds):
     st.write(preds_list)
     
     # Get the name of the group where the prediction is equal to "1"
-    result = [label_dict[key] for key, value in enumerate(preds_list) if value == 1]
+    result = []
+
+    for sublist in my_list_of_lists:
+        result.extend([my_dict[key] for key, value in enumerate(sublist) if value == 1])
 
     st.write("Here are the results")
     st.write(result)