Update app.py

app.py

@@ -19,6 +19,9 @@ st.set_page_config(layout="wide")
 
 clearEphimeral()
 
+if 'wip' not in st.session_state:
+    st.session_state['wip'] = False
+
 detector = pipeline("object-detection", model="sergioska/yolo-finetuned-flyers")
 
 st.title('Upload an image file to detection')
@@ -36,7 +39,7 @@ with col1:
     
     imageLocation = st.empty()
     
-    if uploaded_image_zero_file is not None:
+    if uploaded_image_zero_file is not None and !st.session_state['wip']:
         image = Image.open(uploaded_image_zero_file)
         outputImage = np.array(image)
         predictions = detector(
@@ -45,7 +48,7 @@ with col1:
         )
         imageLocation.image(outputImage)
     
-    if st.button('detect'):
+    if st.button('detect') and !st.session_state['wip']:
     
         draw = ImageDraw.Draw(image)
 
@@ -69,6 +72,7 @@ with col1:
 
         totalGibs = i
         gibsOk = True
+        st.session_state['wip'] = True
         imageLocation.image(image)
         st.write(predictions)