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

app = FastAPI()
classifier = pipeline("text-classification", model="vAlperen/finsight-financial-finbert")

class Input(BaseModel):
    inputs: str

@app.post("/classify")
def classify(body: Input):
    return classifier(body.inputs)

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=7860)