leavoigt commited on
Commit
41aa9dd
1 Parent(s): aaa7178

Update appStore/vulnerability_analysis.py

Browse files
Files changed (1) hide show
  1. appStore/vulnerability_analysis.py +2 -2
appStore/vulnerability_analysis.py CHANGED
@@ -57,7 +57,7 @@ def app():
57
  ### Main app code ###
58
  with st.container():
59
 
60
- if 'key1' in st.session_state:
61
  df = st.session_state.key1
62
  classifier = load_vulnerabilityClassifier(classifier_name=params['model_name'])
63
  st.session_state['{}_classifier'.format(classifier_identifier)] = classifier
@@ -70,7 +70,7 @@ def app():
70
  df = vulnerability_classification(haystack_doc=df,
71
  threshold= params['threshold'])
72
 
73
- st.session_state.key1 = df
74
 
75
 
76
  # # st.write(df)
 
57
  ### Main app code ###
58
  with st.container():
59
 
60
+ if 'key0' in st.session_state:
61
  df = st.session_state.key1
62
  classifier = load_vulnerabilityClassifier(classifier_name=params['model_name'])
63
  st.session_state['{}_classifier'.format(classifier_identifier)] = classifier
 
70
  df = vulnerability_classification(haystack_doc=df,
71
  threshold= params['threshold'])
72
 
73
+ st.session_state.key0 = df
74
 
75
 
76
  # # st.write(df)