Update new_app.py

new_app.py

@@ -24,17 +24,26 @@ def get_model():
     # compile=False skips optimizer/state rebuild, which saves time
     return keras.models.load_model("0.0008-0.92.keras", compile=False)
 
-
 st.markdown("""
-        <style>
-               .block-container {
-                    padding-top: 1rem;
-                    padding-bottom: 0rem;
-                    padding-left: 5rem;
-                    padding-right: 5rem;
-                }
-        </style>
-        """, unsafe_allow_html=True)
+<style>
+.block-container {
+    /* give the header breathing room so it doesn't look clipped */
+    padding-top: 2.2rem;  /* was 1rem */
+    padding-bottom: 0rem; padding-left: 5rem; padding-right: 5rem;
+}
+/* tighten the horizontal rule spacing (bring it closer to the header) */
+hr { 
+    margin-top: 0.25cm;   /* ~2.5mm closer than default */
+    margin-bottom: 0.25cm;
+}
+/* center images if you ever use a fixed width instead of use_container_width */
+.stImage img { 
+    display: block; 
+    margin-left: auto; 
+    margin-right: auto; 
+}
+</style>
+""", unsafe_allow_html=True)
 
 #title
 col1, col2 = st.columns(2)
@@ -80,7 +89,7 @@ if state.img is None:
     except Exception as e:
         st.info(f"Couldn’t fetch default tile: {e}")
 
-col1, col2, col3 = st.columns(3)
+col1, col2, col3 = st.columns([0.8, 1.8, 1.0])  # adjust ratios to taste
 
 with col3:
     #header
@@ -106,7 +115,7 @@ with col2:
             st.error(f"Error fetching image: {e}")
 
     if state.img is not None:
-        st.image(state.img, use_container_width = True)
+        st.image(state.img, width=640) #use_container_width = True
 
 with col1:
     st.subheader("Prediction")