Update app.py

app.py

@@ -40,8 +40,8 @@ with col1:
 
     if not st.session_state['wip']:
         imageLocation = st.empty()
-    #else:
-    #    imageLocation = st.image(st.session_state['image'])
+    else:
+        imageLocation = st.image(st.session_state['image'])
     
     if uploaded_image_zero_file is not None:
         image = Image.open(uploaded_image_zero_file)
@@ -78,7 +78,7 @@ with col1:
         gibsOk = True
         imageLocation.image(image)
         st.session_state['wip'] = True
-        #st.session_state['image'] = image.__dict__
+        st.session_state['image'] = image
         st.write(predictions)
         
 with col2: