Update app.py

app.py

@@ -148,11 +148,13 @@ def main():
             
                     # Convert images to RGB for display
                     new_img = cv2.cvtColor(transformed_chessboard, cv2.COLOR_BGR2RGB)
-                    st.image(new_img, caption="Segmented Chessboard", use_column_width=True)
+                    
 
 
         # Resize the image to 224x224
+        
         image = cv2.resize(new_img, (224, 224))
+        st.image(image, caption="Segmented Chessboard", use_container_width=True)
         height, width, _ = image.shape
 
         # Get user input for perspective