Update app.py

app.py

@@ -109,7 +109,7 @@ if uploaded_files:
             conf = float(np.max(probs))
 
             with cols[j]:
-                st.image(display_img, caption=f"Pred: {label} ({conf*100:.1f}%)", use_column_width=True)
+                st.image(display_img, caption=f"Pred: {label} ({conf*100:.1f}%)", use_container_width=True)
                 st.bar_chart(probs)
 
 # ---- Drawing Pad ----
@@ -141,7 +141,7 @@ if canvas_result is not None and canvas_result.image_data is not None:
             """,
             unsafe_allow_html=True
         )
-        st.image(display_img, caption="Preprocessed (28×28) view", width=120)
+        st.image(display_img, caption="Preprocessed (28×28) view", width=60)
         st.bar_chart(probs)
 
 # ---- Footer ----