Update app.py

app.py

@@ -37,8 +37,10 @@ with col1:
     st.header('flyer')
     uploaded_image_zero_file = st.file_uploader("Choose an image file (zero)")
 
-    if st.session_state['wip']:
+    if not st.session_state['wip']:
         imageLocation = st.empty()
+    else:
+        imageLocation = st.image(st.session_state['image'])
     
     if uploaded_image_zero_file is not None:
         image = Image.open(uploaded_image_zero_file)
@@ -74,7 +76,7 @@ with col1:
         totalGibs = i
         gibsOk = True
         imageLocation.image(image)
-        st.session_state['image'] = image
+        st.session_state['image'] = image.__dict__
         st.write(predictions)
         
 with col2: