Update src/streamlit_app.py

src/streamlit_app.py

@@ -186,6 +186,8 @@ if 'model' not in st.session_state:
 
     st.session_state.started=False
 
+    st.session_state.which_pic=1
+
 
 
 
@@ -258,14 +260,15 @@ if st.session_state.current_image is not None:
 
 
 
-
+        print("before model fit")
         st.session_state.model.fit(image, label, epochs=1)
         print("after model fit")
         #st.session_state.unprocessed_image=None
         print("before fetch_image")
         fetch_image()
         print("after fetch_image")
-        st.info(f"**True Label of last was:** {label_input}")
+        st.info(f"You clicked on last picture (picture {st.session_state.which_pic}): {label_input}")
+        st.session_state.which_pic=st.session_state.which_pic+1