Update main.py

main.py

@@ -4,15 +4,22 @@ from typing import List
 import joblib
 import numpy as np
 
-# Initialize app
-app = FastAPI(title="Random Forest Classifier API")
+app = FastAPI(title="Stroke Risk Prediction API")
 
-# Load the saved model
-model = joblib.load("best_random_forest.pkl")
+# Lazy loading
+model = None
+
+@app.on_event("startup")
+def load_model():
+    global model
+    try:
+        model = joblib.load("best_random_forest.pkl")
+    except Exception as e:
+        print(f"Failed to load model: {e}")
+        raise RuntimeError("Model loading failed.")
 
-# Define input data format
 class PredictRequest(BaseModel):
-    input_data: List[List[float]]  # 2D list for batch prediction
+    input_data: List[List[float]]
 
 @app.post("/predict")
 def predict(req: PredictRequest):
@@ -22,3 +29,11 @@ def predict(req: PredictRequest):
         return {"predictions": predictions.tolist()}
     except Exception as e:
         raise HTTPException(status_code=400, detail=str(e))
+
+@app.get("/")
+def read_root():
+    return {"status": "ok"}
+
+@app.get("/health")
+def health():
+    return {"status": "healthy"}