vAlperen's picture
Create app.py
f317cd9 verified
Raw
History Blame Contribute Delete
417 Bytes
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)