Update app.py

app.py

@@ -197,8 +197,6 @@ if st.session_state.screen == 2:
     st.session_state.tiles = tile_image(image=st.session_state.image, tile_size=640, overlap=40)
 
     with row5_col2:
-        # st.image(st.session_state.image, use_column_width=True)
-
         st.session_state.boxed_image = np.array(st.session_state.image)
         # st.session_state.boxed_image = cv2.cvtColor(st.session_state.boxed_image, cv2.COLOR_RGB2BGR)
         height, width, channels = st.session_state.boxed_image.shape
@@ -242,7 +240,7 @@ if st.session_state.screen == 2:
             st.rerun()
         
         # plot predicted image
-        st.markdown(f"<h4 style='text-align: center; color: black;'>Image {str(st.session_state.image_counter + 1)}</h4>", unsafe_allow_html=True)
+        st.markdown(f"<h4 style='text-align: center; color: black;'>Image Tile {str(st.session_state.image_counter + 1)}</h4>", unsafe_allow_html=True)
         im_bgr = st.session_state.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)