Update utils/vulnerability_classifier.py

utils/vulnerability_classifier.py

@@ -42,7 +42,8 @@ def get_vulnerability_labels(preds):
     result = []
 
     for sublist in preds_list:
-        result.extend([label_dict[key] for key, value in enumerate(sublist) if value == 1])
+        names = [label_dict[key] for key, value in enumerate(sublist) if value == 1])
+        result.append(names)
 
     st.write("Here are the results")
     st.write(result)