Update app.py

app.py

@@ -200,12 +200,12 @@ if st.session_state.screen == 2:
         # 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)
+        # st.session_state.boxed_image = cv2.cvtColor(st.session_state.boxed_image, cv2.COLOR_RGB2BGR)
         height, width, channels = st.session_state.boxed_image.shape
         st.session_state.blank_img = np.zeros((height + 640, width + 640, 3), np.uint8)
         st.session_state.blank_img[0:height, 0:width] = st.session_state.boxed_image
-        for i in range(0, height, 600):
-            for j in range(0, width, 600):
+        for i in range(0, width, 600):
+            for j in range(0, height, 600):
                 cv2.rectangle(st.session_state.blank_img, (j, i), (j + 640, i + 640), (0, 255, 0), 5)
                 cv2.putText(st.session_state.blank_img, str(int(i / 600) * 10 + int(j / 600)), (j + 200, i + 320), cv2.FONT_HERSHEY_SIMPLEX, 10, (0, 255, 0), 10)        
         st.image(st.session_state.blank_img, use_column_width=True)