Update app.py

app.py

@@ -152,6 +152,7 @@ def load_CADET():
 def reset_threshold():
     # del st.session_state["threshold"]
     st.session_state['threshold'] = 0.0
+    st.session_state['example'] = False
 
 
 # Load model
@@ -216,7 +217,7 @@ if uploaded_file is not None:
     os.system(f'mkdir -p {fname}')
     data, wcs = load_file(uploaded_file)
 
-if example:
+if st.session_state.get("example", True):
     fname = "NGC4649_example"
     os.system(f'mkdir -p {fname}')
     data, wcs = load_file(f"{fname}.fits")
@@ -253,7 +254,7 @@ with col5: decompose = st.button('Decompose', key="decompose")
 # Make two columns for plots
 _, colA, colB, colC, _ = st.columns([bordersize,1,1,1,bordersize])
 
-if (uploaded_file is not None) or st.session_state.get("example", False):
+if (uploaded_file is not None) or st.session_state.get("example", True):
     # NORMALIZE IMAGE
     MIN = np.min(np.where(data == 0, 1, data))
     if MIN < 1: data = data / MIN