Update app.py

app.py

@@ -50,11 +50,15 @@ if uploaded_image:
     image = Image.open(uploaded_image).convert("RGB")
     st.image(image, caption="Original Image", use_column_width=True)
 
-    # Preprocess and predict
+    # Preprocess and predict (NO normalization here)
     resized = image.resize((512, 512))
-    x = np.expand_dims(np.array(resized) / 255.0, axis=0)
+    x = np.expand_dims(np.array(resized), axis=0)
     y = model.predict(x)[0, :, :, 0]
 
+    # Print min/max to logs
+    st.text(f"Prediction min/max: {y.min():.5f} / {y.max():.5f}")
+
+    # Contrast stretch
     y_norm = (y - y.min()) / (y.max() - y.min() + 1e-6)
     mask = (y_norm * 255).astype(np.uint8)
     result = Image.fromarray(mask)