Update app.py

app.py

@@ -47,10 +47,12 @@ if uploaded_files:
                 image = Image.open(uploaded_file).convert("RGB")
                 preview = image.copy()
                 preview.thumbnail((150, 150))
-                st.image(preview, caption="Preview", use_container_width=True)
+                st.image(preview, caption="📷 Preview", use_container_width=True)
 
                 # 🧪 Preprocess image
                 resized = image.resize((128, 128))  # Adjust size as per your model
+                st.image(resized, caption="🧪 Resized for Model (128x128)", use_container_width=False)
+
                 img_array = img_to_array(resized) / 255.0
                 img_array = np.expand_dims(img_array, axis=0)
 
@@ -68,6 +70,8 @@ if uploaded_files:
 
             except UnidentifiedImageError:
                 st.error("❌ Error: Invalid image file.")
+            except Exception as e:
+                st.error(f"⚠️ Unexpected error: {str(e)}")
 
 # 📌 Footer message
 st.sidebar.markdown("---")