Update app.py

app.py

@@ -77,6 +77,7 @@ with col1:
         totalGibs = i
         gibsOk = True
         imageLocation.image(image)
+        st.session_state['wip'] = True
         st.session_state['image'] = image.__dict__
         st.write(predictions)