from fastapi import FastAPI
import pandas as pd
import joblib

app = FastAPI()

# Cargar el modelo entrenado
model = joblib.load("model.joblib")

@app.post("/predict")
def predict(data: dict):
    df = pd.DataFrame([data])
    df = pd.get_dummies(df)
    prediction = model.predict(df)
    return {"prediction": prediction[0]}