Update main.py

main.py

@@ -1,26 +1,26 @@
-from fastapi import FastAPI, HTTPException, Request
-import os
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+from typing import List
+import numpy as np
+
+# import the pure‑Python scoring function
 from rf_model import score
 
 app = FastAPI(title="Stroke Risk (Pure‑Python RF)")
 
-@app.get("/")
-def root():
-    return {"status": "ok"}
+class PredictRequest(BaseModel):
+    input_data: List[List[float]]
 
 @app.get("/health")
 def health():
     return {"status": "healthy"}
 
 @app.post("/predict")
-async def predict(request: Request):
+def predict(req: PredictRequest):
     try:
-        body = await request.json()
-        data = body.get("input_data")
-        if not isinstance(data, list):
-            raise ValueError("`input_data` must be a list of rows")
-        # each row is a list of floats
-        preds = [score(row) for row in data]
+        arr = np.array(req.input_data, dtype=float)
+        # call the generated function directly
+        preds = [score(row.tolist()) for row in arr]
         return {"predictions": preds}
     except Exception as e:
         raise HTTPException(status_code=400, detail=str(e))