Update utils/vulnerability_classifier.py

utils/vulnerability_classifier.py

@@ -37,12 +37,14 @@ def get_vulnerability_labels(preds):
     
     # Get label names
     preds_list = preds.tolist()
-    st.write("preds_list")
+    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]
 
     st.write("Here are the results")
     st.write(result)
+    
     return result
         
 @st.cache_resource