Update stroke-flask-docker/app.py

stroke-flask-docker/app.py

@@ -2,6 +2,7 @@ from flask import Flask, render_template, request, jsonify
 import joblib
 import numpy as np
 import os
+import pandas as pd  
 
 APP_PORT = int(os.getenv("PORT", "8080"))
 
@@ -49,7 +50,6 @@ def index():
         "smoking_status": "never smoked",
     }
     return render_template("index.html", defaults=defaults)
-
 @app.route("/predict", methods=["POST"])
 def predict():
     try:
@@ -67,12 +67,11 @@ def predict():
 
         # ALWAYS send a DataFrame with named columns
         X = pd.DataFrame([{f: payload.get(f, None) for f in FEATURE_ORDER}])[FEATURE_ORDER]
-
+        prob = float(pipeline.predict_proba(X)[0][1])
         # quick sanity log (optional)
         # print("X type:", type(X), "cols:", list(X.columns))
 
-        prob = float(pipeline.predict_proba(X)[0][1])
-        pred = int(prob >= 0.5)
+        pred = int(prob >= 0.3)
         result = {"stroke_probability": prob, "predicted_label": pred}
 
         return jsonify(result) if request.is_json else render_template("index.html", result=result, defaults=payload)
@@ -80,6 +79,6 @@ def predict():
     except Exception as e:
         return (jsonify({"error": str(e)}), 400) if request.is_json else \
                (render_template("index.html", error=str(e), defaults=payload), 400)
-        
+
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=APP_PORT, debug=False)