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_container_width=True)
+                st.image(display_img, caption=f"Pred: {label} ({conf*100:.1f}%)", width=60)
                 st.bar_chart(probs)
 
 # ---- Drawing Pad ----