Update app.py

app.py

@@ -22,6 +22,7 @@ clearEphimeral()
 
 if 'wip' not in st.session_state:
     st.session_state['wip'] = False
+    st.session_state['image'] = None
 
 detector = pipeline("object-detection", model="sergioska/yolo-finetuned-flyers")
 
@@ -35,8 +36,12 @@ with col1:
 
     st.header('flyer')
     uploaded_image_zero_file = st.file_uploader("Choose an image file (zero)")
-    
-    imageLocation = st.empty()
+
+    if not st.session_state['wip']:
+        imageLocation = st.empty()
+    else:
+        image = st.session_state['image']
+        imageLocation.image(image)
     
     if uploaded_image_zero_file is not None:
         image = Image.open(uploaded_image_zero_file)
@@ -72,6 +77,7 @@ with col1:
         totalGibs = i
         gibsOk = True
         imageLocation.image(image)
+        st.session_state['image'] = image
         st.write(predictions)
         
 with col2: