toddmattingly's picture
Upload main.py
fc842d5 verified
raw
history blame contribute delete
438 Bytes
from fastapi import FastAPI, Request
import joblib
import pandas as pd
app = FastAPI()
model = joblib.load("model.joblib")
@app.get("/")
def home():
return {"status": "ok", "model_loaded": True, "version": "v4.0.3"}
@app.post("/predict")
async def predict(req: Request):
data = await req.json()
df = pd.DataFrame(data)
preds = model.predict(df)
return {"status": "ok", "predictions": preds.tolist()}