Update app.py

app.py

@@ -13,7 +13,6 @@ def load_model():
 model = load_model()
 
 st.title("🫁 Pneumonia Detection from Chest X-ray Images")
-
 st.markdown("Upload your own X-ray or try one of the sample images below.")
 
 # === Sample Image Section ===
@@ -43,9 +42,12 @@ if image and st.button('🔍 Predict'):
     img_array = np.array(img) / 255.0
     img_array = np.expand_dims(img_array, axis=0)
 
-    prediction = model.predict(img_array)
+    prediction = model.predict(img_array)[0][0]
 
-    if prediction[0][0] > 0.5:
+    # Set thresholds to determine confidence
+    if prediction < 0.3:
+        st.success("✅ **Prediction: Normal**")
+    elif prediction > 0.7:
         st.error("🩺 **Prediction: Pneumonia Detected**")
     else:
-        st.success("✅ **Prediction: Normal**")
+        st.warning("⚠️ Unable to confidently detect pneumonia. Please upload a clear chest X-ray image.")