Solve Jitter Bug

app.py

@@ -231,7 +231,7 @@ with col2:
             # Display Side by Side (Input vs Prediction)
             res_col1, res_col2 = st.columns(2)
             with res_col1:
-                st.image(transformed_img, caption=f"Original: '{captcha_text}'", use_container_width=True)
+                st.image(transformed_img, caption=f"Original: '{captcha_text}'")
             
             with res_col2:
                 # Character-level coloring
@@ -274,18 +274,19 @@ with col_sec2_1:
         "Value": list(stats_2.values())
     })
 
+with col_sec2_2:
+    st.subheader("Upload an image")
+    
     with st.expander("Show Example Images"):
-        st.markdown("Drag and Drop one of these images into the uploader above!")
+        st.markdown("Drag and Drop one of these images into the uploader below!")
         
         # Display in a grid of 3 columns
-        with st.container(height=400):
+        with st.container(height=300):
             cols = st.columns(3)
             for i, img_path in enumerate(image_files):
                 with cols[i % 3]:
                     st.image(img_path, use_container_width=True)
 
-with col_sec2_2:
-    st.subheader("Upload an image")
     uploaded_file = st.file_uploader("Choose an image file", type=["png", "jpg", "jpeg"], key="test_uploader")
                     
     image_to_predict = None