vulnerability_2_1

Sleeping

leavoigt commited on Sep 25, 2023

Commit

d7db62f

•

1 Parent(s): f6a2493

Update utils/vulnerability_classifier.py

Files changed (1) hide show

utils/vulnerability_classifier.py CHANGED Viewed

@@ -68,7 +68,7 @@ def load_Classifier(config_file:str = None, classifier_name:str = None):
 @st.cache(allow_output_mutation=True)
-def classification(haystack_doc:List[Document],
                         threshold:float = 0.8,
                         classifier_model:TransformersDocumentClassifier= None
                         )->Tuple[DataFrame,Series]:

 @st.cache(allow_output_mutation=True)
+def vulnerability_classification(haystack_doc:List[Document],
                         threshold:float = 0.8,
                         classifier_model:TransformersDocumentClassifier= None
                         )->Tuple[DataFrame,Series]: