app.py

from transformers import pipeline
from fastapi import FastAPI

app = FastAPI()
classifier = pipeline("text-classification", model="KeivanR/Qwen2.5-1.5B-Instruct-MLB-clf_lora-1743189446")

@app.post("/predict")
def predict(text: str):
    return classifier(text)