vulnerability_2_1

Sleeping

leavoigt commited on Jan 4, 2024

Commit

cdc5cd2

1 Parent(s): 2decf42

Update utils/vulnerability_classifier.py

Files changed (1) hide show

utils/vulnerability_classifier.py CHANGED Viewed

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

     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