Reshape Image

app.py

@@ -48,12 +48,12 @@ def download_comparison_images():
 download_comparison_images()
 
 if "output_1" not in st.session_state:
-    image_1 = Image.open('plant_pid.png')
-    st.session_state["output_1"] = image_1.thumbnail((4960, 3508))
+    with Image.open('plant_pid.png') as img_1:
+        st.session_state["output_1"] = img_1.load
 
 if "output_2" not in st.session_state:
-    image_2 = Image.open('prediction_visual.png')
-    st.session_state["output_2"] = image_2.thumbnail((4960, 3508))
+    with Image.open('prediction_visual.png') as img_2:
+        st.session_state["output_2"] = img_2.load
 
 col1, col2, col3 = st.columns(3, gap='medium')
 with col1: