Update src/streamlit_app.py

src/streamlit_app.py

@@ -198,35 +198,36 @@ if 'model' not in st.session_state:
 
     st.session_state.which_pic=1
 
-    st.session_state.calculating=False
+    st.session_state.calculating=True
 
 
     
 # Button to fetch a new image
 col1, col2, col3 = st.columns([1,1,1])
 
-if st.session_state.current_image is not None and st.session_state.started is False:
-    st.session_state.started=True
 
 with col1: 
-    if st.button("Start",key="started",disabled=st.session_state.started):
+    if st.button("Start",disabled=st.session_state.started):
         if not st.session_state.started:
             fetch_image()
-            st.button("Start",key="started",disabled=True)
+            st.session_state.started=True
+            st.session_state.calculating=False
+            
             
 
 
 # Display the current image
 
+
+with col2:
+    if st.button("cat",disabled=st.session_state.calculating):
+        print("cat pressed")
+        st.session_state.label_input="cat"
+with col3:
+    if st.button("dog",disabled=st.session_state.calculating):
+        print("dog pressed")
+        st.session_state.label_input="dog"
 if st.session_state.current_image is not None:
-    with col2:
-        if st.button("cat",disabled=st.session_state.calculating):
-            print("cat pressed")
-            st.session_state.label_input="cat"
-    with col3:
-        if st.button("dog",disabled=st.session_state.calculating):
-            print("dog pressed")
-            st.session_state.label_input="dog"
     print(f"SHAPE:{st.session_state.current_image.shape}")
     prediction = st.session_state.model.predict(np.array([st.session_state.current_image]))[0][0]
 
@@ -291,7 +292,8 @@ if st.session_state.calculating:
     print("after fetch_image")
     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
-    st.session_state.calculating=False
+    
+    st.session_state.started=False