session fixed

app.py

@@ -131,13 +131,16 @@ st.markdown(
 # introduce states
 if "prev_image" not in st.session_state:
     st.session_state.prev_image = None 
+if "reset_model" not in st.session_state:
+    st.session_state.reset_model = False
 if "model_key" not in st.session_state:
     st.session_state.model_key = "default_model_key"
 
 user_image = st.file_uploader("png, jpg, or jpeg image", ['png', 'jpg', 'jpeg'], label_visibility='hidden')
 if user_image != st.session_state.prev_image:
     if st.session_state.prev_image is not None: 
-        st.session_state.model_key = "reset_model_key"  # reset after the first upload
+        st.session_state.model_key = "reset_model_key" if st.session_state.model_key == "default_model_key" else "default_model_key"
+        st.session_state.reset_model = True
     st.session_state.prev_image = user_image  # set prev image to current image 
 
 model_name = st.selectbox(