Reshape Image

app.py

@@ -49,11 +49,11 @@ download_comparison_images()
 
 if "output_1" not in st.session_state:
     with Image.open('plant_pid.png') as img_1:
-        st.session_state["output_1"] = img_1.load
+        st.session_state["output_1"] = img_1.load()
 
 if "output_2" not in st.session_state:
     with Image.open('prediction_visual.png') as img_2:
-        st.session_state["output_2"] = img_2.load
+        st.session_state["output_2"] = img_2.load()
 
 col1, col2, col3 = st.columns(3, gap='medium')
 with col1: