from fastapi import FastAPI
from pydantic import BaseModel
from transformers import pipeline

app = FastAPI()

# Load your model
clf = pipeline(
    "text-classification",
    model="DelaliScratchwerk/time-period-classifier-bert"
)

class PredictRequest(BaseModel):
    inputs: str

@app.post("/predict")
def predict(req: PredictRequest):
    result = clf(req.inputs)
    return result