Spaces:
Sleeping
Sleeping
Update utils/vulnerability_classifier.py
Browse files
utils/vulnerability_classifier.py
CHANGED
@@ -68,7 +68,7 @@ def load_Classifier(config_file:str = None, classifier_name:str = None):
|
|
68 |
|
69 |
|
70 |
@st.cache(allow_output_mutation=True)
|
71 |
-
def
|
72 |
threshold:float = 0.8,
|
73 |
classifier_model:TransformersDocumentClassifier= None
|
74 |
)->Tuple[DataFrame,Series]:
|
|
|
68 |
|
69 |
|
70 |
@st.cache(allow_output_mutation=True)
|
71 |
+
def vulnerability_classification(haystack_doc:List[Document],
|
72 |
threshold:float = 0.8,
|
73 |
classifier_model:TransformersDocumentClassifier= None
|
74 |
)->Tuple[DataFrame,Series]:
|