app.py

from fastapi import FastAPI
from transformers import pipeline

app = FastAPI()

# Cargar el modelo DistilBERT para clasificación de texto
classifier = pipeline("text-classification", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")

@app.get("/")
def greet_json():
    return {"Hello": "World!"}

@app.post("/predict/")
def predict(text: str):
    result = classifier(text)
    return {"label": result[0]['label'], "score": result[0]['score']}