Update app.py

app.py

@@ -39,7 +39,7 @@ with col1:
     
     imageLocation = st.empty()
     
-    if uploaded_image_zero_file is not None and not st.session_state['wip']:
+    if uploaded_image_zero_file is not None:
         image = Image.open(uploaded_image_zero_file)
         outputImage = np.array(image)
         predictions = detector(
@@ -48,7 +48,7 @@ with col1:
         )
         imageLocation.image(outputImage)
     
-    if st.button('detect') and not st.session_state['wip']:
+    if st.button('detect'):
     
         draw = ImageDraw.Draw(image)
 
@@ -72,7 +72,6 @@ with col1:
 
         totalGibs = i
         gibsOk = True
-        st.session_state['wip'] = True
         imageLocation.image(image)
         st.write(predictions)