Update app.py

app.py

@@ -154,7 +154,10 @@ def reset_threshold():
     st.session_state['threshold'] = 0.0
     # st.session_state['example'] = False
 
-
+#@st.cache #_data
+def set_example():
+    st.session_state['example'] = True
+    
 # Load model
 model = load_CADET()
 
@@ -194,7 +197,7 @@ with col_1:
 
 with col_2:
     st.markdown("<br style='margin:15px 0'>", unsafe_allow_html=True)
-    example = st.button("Example", key="example")
+    example = st.button("Example", on_click=set_example)
     
     # with col_2:
     #     st.markdown("### Examples")
@@ -219,7 +222,7 @@ if uploaded_file is not None:
     MIN = np.min(np.where(data == 0, 1, data))
     if MIN < 1: data = data / MIN
 
-if st.session_state.get("example", True):
+if st.session_state["example"]:
     fname = "NGC4649_example"
     os.system(f'mkdir -p {fname}')
     data, wcs = load_file(f"{fname}.fits")