Update app.py

app.py

@@ -62,38 +62,38 @@ with row2_col2:
     st.image(img_1, use_column_width=True)
     if st.button(f"Select Image ↑", key="button_1"):
         st.session_state.image = img_1
-with row3_col2:
-    img_2 = Image.open(image_path + "IMG_5382.JPG")
+with row2_col3:
+    img_2 = Image.open(image_path + "IMG_5357.JPG")
     img_2 = ImageOps.exif_transpose(img_2)
     st.image(img_2, use_column_width=True)
     if st.button(f"Select Image ↑", key="button_2"):
         st.session_state.image = img_2
-with row2_col3:
-    img_3 = Image.open(image_path + "IMG_5357.JPG")
+with row2_col4:
+    img_3 = Image.open(image_path + "IMG_5368.JPG")
     img_3 = ImageOps.exif_transpose(img_3)
     st.image(img_3, use_column_width=True)
     if st.button(f"Select Image ↑", key="button_3"):
         st.session_state.image = img_3
-with row3_col3:
-    img_4 = Image.open(image_path + "IMG_5383.JPG")
+with row2_col5:
+    img_4 = Image.open(image_path + "IMG_5369.JPG")
     img_4 = ImageOps.exif_transpose(img_4)
     st.image(img_4, use_column_width=True)
     if st.button(f"Select Image ↑", key="button_4"):
         st.session_state.image = img_4
-with row2_col4:
-    img_5 = Image.open(image_path + "IMG_5368.JPG")
+with row3_col2:
+    img_5 = Image.open(image_path + "IMG_5382.JPG")
     img_5 = ImageOps.exif_transpose(img_5)
     st.image(img_5, use_column_width=True)
     if st.button(f"Select Image ↑", key="button_5"):
         st.session_state.image = img_5
-with row3_col4:
-    img_6 = Image.open(image_path + "IMG_5408.JPG")
+with row3_col3:
+    img_6 = Image.open(image_path + "IMG_5383.JPG")
     img_6 = ImageOps.exif_transpose(img_6)
     st.image(img_6, use_column_width=True)
     if st.button(f"Select Image ↑", key="button_6"):
         st.session_state.image = img_6
-with row2_col5:
-    img_7 = Image.open(image_path + "IMG_5369.JPG")
+with row3_col4:
+    img_7 = Image.open(image_path + "IMG_5408.JPG")
     img_7 = ImageOps.exif_transpose(img_7)
     st.image(img_7, use_column_width=True)
     if st.button(f"Select Image ↑", key="button_7"):
@@ -149,15 +149,15 @@ if st.session_state.image is not None:
         row4a_col3, row4b_col3= st.columns([1,1], gap="medium")
         with row4a_col3:
             if st.button("Back"):
-                if session_state.image_counter > 0:
-                    session_state.image_counter -= 1
+                if st.session_state.image_counter > 0:
+                    st.session_state.image_counter -= 1
         with row4b_col3:
             if st.button("Next"):
-                if session_state.image_counter < len(results):
-                    session_state.image_counter -= 1
+                if st.session_state.image_counter < len(results):
+                    st.session_state.image_counter -= 1
                 
         # Plot predicted image
-        im_bgr = results[session_state.image_counter].plot()  # BGR-order numpy array
+        im_bgr = results[st.session_state.image_counter].plot()  # BGR-order numpy array
         im_rgb = Image.fromarray(im_bgr[..., ::-1])  # RGB-order PIL image
         st.image(im_rgb)