Update app.py

app.py

@@ -148,11 +148,14 @@ def load_CADET():
     # model = load_model("CADET.hdf5")
     return model
 
+def reset_all():
+    # del st.session_state["threshold"]
+    st.session_state['threshold'] = 0.0
+    st.session_state['example'] = False
 
 def reset_threshold():
     # del st.session_state["threshold"]
     st.session_state['threshold'] = 0.0
-    st.session_state['example'] = False
     
 # Load model
 model = load_CADET()
@@ -189,7 +192,7 @@ If you use this tool in your research, please cite <a href='https://arxiv.org/ab
 _, col_1, col_2, _ = st.columns([bordersize, 2.5, 0.5, bordersize])
 
 with col_1:
-    uploaded_file = st.file_uploader("Choose a FITS file", type=['fits'], on_change=reset_threshold)
+    uploaded_file = st.file_uploader("Choose a FITS file", type=['fits'], on_change=reset_all)
 
 with col_2:
     st.markdown("<br style='margin:15px 0'>", unsafe_allow_html=True)