main.py

from fastapi import FastAPI, Request
from pydantic import BaseModel
from app.model_utils import load_model, predict_label

app = FastAPI()
tokenizer, model, label_encoders = load_model()

class InputText(BaseModel):
    text: str

@app.post("/predict")
def predict(input: InputText):
    predictions = predict_label(input.text, tokenizer, model, label_encoders)
    return predictions