Update src/streamlit_app.py

src/streamlit_app.py

@@ -167,6 +167,8 @@ if 'model' not in st.session_state:
 
     st.session_state.label_input = None
 
+    st.session_state.started=False
+
 
 def fetch_image():
     print("entering fetch_image")
@@ -189,9 +191,7 @@ def fetch_image():
 # Button to fetch a new image
 col1, col2, col3 = st.columns([1,1,1])
 
-if st.session_state.current_image is None:
-    st.session_state.started=False
-else:
+if st.session_state.current_image is not None and st.session_state.started is False:
     st.session_state.started=True
 
 with col1: 
@@ -259,9 +259,9 @@ if st.session_state.current_image is not None:
 
 
         st.session_state.model.fit(image, label, epochs=1, verbose=0)
-        #st.session_state.unprocessed_image=None
+        st.session_state.unprocessed_image=None
         fetch_image()
-        st.rerun()
+        #st.rerun()