Update appStore/vulnerability_analysis.py

appStore/vulnerability_analysis.py

@@ -57,7 +57,7 @@ def app():
     ### Main app code ###
     with st.container():
                    
-            if 'key1' in st.session_state:
+            if 'key0' in st.session_state:
                 df = st.session_state.key1
                 classifier = load_vulnerabilityClassifier(classifier_name=params['model_name'])
                 st.session_state['{}_classifier'.format(classifier_identifier)] = classifier
@@ -70,7 +70,7 @@ def app():
                 df = vulnerability_classification(haystack_doc=df,
                                             threshold= params['threshold'])
 
-                st.session_state.key1 = df
+                st.session_state.key0 = df
 
 
                 # # st.write(df)