Update utils/vulnerability_classifier.py

utils/vulnerability_classifier.py

@@ -85,7 +85,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(classifier_name)
+    doc_classifier = SetFitModel.from_pretrained("leavoigt/vulnerability_multilabel")
     
     # doc_classifier = pipeline("text-classification", 
     #                         model=classifier_name,