Update appStore/vulnerability_analysis.py

appStore/vulnerability_analysis.py

@@ -62,10 +62,10 @@ def app():
                 classifier = load_vulnerabilityClassifier(classifier_name=params['model_name'])
                 st.session_state['{}_classifier'.format(classifier_identifier)] = classifier
 
-                if sum(df['Target Label'] == 'TARGET') > 100:
-                    warning_msg = ": This might take sometime, please sit back and relax."
-                else:
-                    warning_msg = ""
+                # if sum(df['Target Label'] == 'TARGET') > 100:
+                #     warning_msg = ": This might take sometime, please sit back and relax."
+                # else:
+                #     warning_msg = ""
                     
                 df = vulnerability_classification(haystack_doc=df,
                                             threshold= params['threshold'])