leavoigt commited on
Commit
d7db62f
1 Parent(s): f6a2493

Update utils/vulnerability_classifier.py

Browse files
Files changed (1) hide show
  1. utils/vulnerability_classifier.py +1 -1
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 classification(haystack_doc:List[Document],
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]: