Update app.py

app.py

@@ -153,10 +153,6 @@ def reset_threshold():
     # del st.session_state["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()
@@ -213,6 +209,10 @@ with col_2:
     #     uploaded_file = "NGC5813_example.fits"
 
     # If file is uploaded, read in the data and plot it
+
+
+if 'example' not in st.session_state:
+    st.session_state['example'] = False
     
 if uploaded_file is not None:
     fname = uploaded_file.name.strip(".fits")
@@ -226,6 +226,7 @@ if st.session_state["example"]:
     fname = "NGC4649_example"
     os.system(f'mkdir -p {fname}')
     data, wcs = load_file(f"{fname}.fits")
+    st.session_state['example'] = True
 
 if "data" not in locals():
     data, wcs = np.ones((128,128)) * (-1), None