Spaces:

TeresaK
/

cpv_test

Runtime error

leavoigt commited on Dec 20, 2023

Commit

3c6fce1

•

1 Parent(s): 185893f

Update utils/vulnerability_classifier.py

Files changed (1) hide show

utils/vulnerability_classifier.py CHANGED Viewed

@@ -88,7 +88,7 @@ def load_vulnerabilityClassifier(config_file:str = None, classifier_name:str = N
     #                     top_k = None)
     # # Download model from HF Hub
-    doc_classifier = SetFitModel.from_pretrained("leavoigt/vulnerable_groups")
     # doc_classifier = pipeline("text-classification",
     #                         model=classifier_name,

     #                     top_k = None)
     # # Download model from HF Hub
+    doc_classifier = SetFitModel.from_pretrained("leavoigt/vulnerability_multilabel")
     # doc_classifier = pipeline("text-classification",
     #                         model=classifier_name,