Update app.py

app.py

@@ -77,6 +77,9 @@ if uploaded_image:
 
     st.image(image, caption="Input Image", use_column_width=True)
 
+    # Confidence threshold slider
+    threshold = st.slider("Confidence Threshold", 0.0, 1.0, 0.5, step=0.01)
+
     # Preprocess and predict
     resized = image.resize((512, 512))
     x = np.expand_dims(np.array(resized), axis=0)
@@ -84,9 +87,8 @@ if uploaded_image:
 
     st.text(f"Prediction min/max: {y.min():.5f} / {y.max():.5f}")
 
-    # Normalize output
-    y_norm = (y - y.min()) / (y.max() - y.min() + 1e-6)
-    mask = (y_norm * 255).astype(np.uint8)
-    result = Image.fromarray(mask)
+    # Apply thresholding if needed
+    mask_bin = (y > threshold).astype(np.uint8) * 255
+    mask_image = Image.fromarray(mask_bin)
 
-    st.image(result, caption="Predicted Segmentation", use_column_width=True)
+    st.image(mask_image, caption=f"Segmentation Mask (Threshold = {threshold:.2f})", use_column_width=True)