Update app.py

app.py

@@ -17,8 +17,16 @@ def predict_alzheimers(age, gender, bmi, smoking, alcohol, diabetes, hypertensio
     apoe4 = 1 if apoe4 == "Yes" else 0
 
     features = np.array([[age, gender, bmi, smoking, alcohol, diabetes, hypertension, family_history, cognitive_score, apoe4]])
-    prediction = model.predict(features)[0]
-    return "🧠 Likely Alzheimer’s Positive" if prediction == 1 else "✅ Likely Alzheimer’s Negative"
+
+    # Get probability of Alzheimer's positive (class 1)
+    proba = model.predict_proba(features)[0][1]
+    proba_percent = round(proba * 100, 2)
+
+    if proba >= 0.5:
+        return f"🧠 Likely Alzheimer’s Positive ({proba_percent}%)"
+    else:
+        return f"✅ Likely Alzheimer’s Negative ({100 - proba_percent}%)"
+
 
 # Gradio UI
 interface = gr.Interface(