leavoigt commited on
Commit
efafdfc
1 Parent(s): 154ee8f

Update appStore/vulnerability_analysis.py

Browse files
Files changed (1) hide show
  1. appStore/vulnerability_analysis.py +4 -4
appStore/vulnerability_analysis.py CHANGED
@@ -62,10 +62,10 @@ def app():
62
  classifier = load_vulnerabilityClassifier(classifier_name=params['model_name'])
63
  st.session_state['{}_classifier'.format(classifier_identifier)] = classifier
64
 
65
- if sum(df['Target Label'] == 'TARGET') > 100:
66
- warning_msg = ": This might take sometime, please sit back and relax."
67
- else:
68
- warning_msg = ""
69
 
70
  df = vulnerability_classification(haystack_doc=df,
71
  threshold= params['threshold'])
 
62
  classifier = load_vulnerabilityClassifier(classifier_name=params['model_name'])
63
  st.session_state['{}_classifier'.format(classifier_identifier)] = classifier
64
 
65
+ # if sum(df['Target Label'] == 'TARGET') > 100:
66
+ # warning_msg = ": This might take sometime, please sit back and relax."
67
+ # else:
68
+ # warning_msg = ""
69
 
70
  df = vulnerability_classification(haystack_doc=df,
71
  threshold= params['threshold'])