modif thresholds

app.py

@@ -124,9 +124,9 @@ def predict_risk(
     diab_prob = float(model_diab.predict_proba(row_diab[feature_cols])[0][1])
     chol_prob = float(model_chol.predict_proba(row_chol[feature_cols])[0][1])
 
-    # Hard classification using threshold 0.5 (same as in notebooks 11 & 12)
-    diab_class = int(diab_prob >= 0.5)
-    chol_class = int(chol_prob >= 0.5)
+    # Hard classification using threshold 0.40 for diab, 0.35 for chol
+    diab_class = int(diab_prob >= 0.40)
+    chol_class = int(chol_prob >= 0.35)
 
     # Gradio HighlightedText needs a list of (text, label/color)
     diab_text = f"Diabetes risk: {diab_prob * 100:.1f}% — Class {diab_class}"
@@ -150,7 +150,7 @@ def predict_risk(
 with gr.Blocks(theme=theme, title="NHANES Health Risk Predictor") as demo:
     gr.Markdown(
         "# NHANES Health Risk Predictor\n"
-        "### Two ML models trained on 7,000+ NHANES participants (2021–2023)\n"
+        "### Two ML models trained on 5,000+ NHANES participants (2021–2023)\n"
         "---"
     )