vulnerability

Sleeping

leavoigt commited on Sep 26, 2023

Commit

785df60

1 Parent(s): b758017

Update utils/vulnerability_classifier.py

Files changed (1) hide show

utils/vulnerability_classifier.py CHANGED Viewed

@@ -52,7 +52,7 @@ def getlabels(preds):
       # Append name to list
       predictions_names.append(name)
-      return predictions_names
 @st.cache_resource
 def load_vulnerabilityClassifier(config_file:str = None, classifier_name:str = None):

       # Append name to list
       predictions_names.append(name)
+    return predictions_names
 @st.cache_resource
 def load_vulnerabilityClassifier(config_file:str = None, classifier_name:str = None):