Update utils/vulnerability_classifier.py
Browse files
utils/vulnerability_classifier.py
CHANGED
|
@@ -136,10 +136,8 @@ def vulnerability_classification(haystack_doc:pd.DataFrame,
|
|
| 136 |
|
| 137 |
|
| 138 |
pred_labels = getlabels(predictions)
|
| 139 |
-
|
| 140 |
-
|
| 141 |
-
st.write(len(haystack_doc))
|
| 142 |
-
# haystack_doc['Vulnerability Label'] = pred_labels
|
| 143 |
# placeholder = {}
|
| 144 |
# for j in range(len(temp)):
|
| 145 |
# placeholder[temp[j]['label']] = temp[j]['score']
|
|
|
|
| 136 |
|
| 137 |
|
| 138 |
pred_labels = getlabels(predictions)
|
| 139 |
+
|
| 140 |
+
haystack_doc['Vulnerability Label'] = pred_labels
|
|
|
|
|
|
|
| 141 |
# placeholder = {}
|
| 142 |
# for j in range(len(temp)):
|
| 143 |
# placeholder[temp[j]['label']] = temp[j]['score']
|