removed function to store whether submit button is clicked in nn tab

app.py

@@ -12,9 +12,6 @@ from streamlit_tags import st_tags, st_tags_sidebar
 
 st.set_page_config(page_title="Ancient Greek Word2Vec", layout="centered")
 
-def click_nn_button():
-    st.session_state.nearest_neighbours = not st.session_state.nearest_neighbours
-
 # Horizontal menu
 active_tab = option_menu(None, ["Nearest neighbours", "Cosine similarity", "3D graph", 'Dictionary'], 
     menu_icon="cast", default_index=0, orientation="horizontal")
@@ -44,7 +41,7 @@ if active_tab == "Nearest neighbours":
             )
         n = st.slider("Number of neighbours", 1, 50, 15)
         
-        nearest_neighbours_button = st.form_submit_button("Find nearest neighbours", on_click = click_nn_button)
+        nearest_neighbours_button = st.form_submit_button("Find nearest neighbours")
         
     if nearest_neighbours_button:
         if validate_nearest_neighbours(target_word, n, models) == False: