Update utils/vulnerability_classifier.py

utils/vulnerability_classifier.py

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