from fastapi import FastAPI
import joblib
import pandas as pd

app = FastAPI()

model = joblib.load("rf_classifer.joblib")

@app.post("/predict")
def predict(data: dict):
    df = pd.DataFrame([data])

    prediction = model.predict(df)

    return {"prediction" : prediction.tolist()}