Update utils/vulnerability_classifier.py

utils/vulnerability_classifier.py

@@ -136,10 +136,8 @@ def vulnerability_classification(haystack_doc:pd.DataFrame,
         
 
         pred_labels = getlabels(predictions)
-        st.write(len(pred_labels))
-        st.write(pred_labels)
-        st.write(len(haystack_doc))
-        # haystack_doc['Vulnerability Label'] = pred_labels
+      
+        haystack_doc['Vulnerability Label'] = pred_labels
     #   placeholder = {}
     #   for j in range(len(temp)):
     #     placeholder[temp[j]['label']] = temp[j]['score']