|
|
from fastapi import FastAPI |
|
|
from pydantic import BaseModel |
|
|
from transformers import pipeline |
|
|
import os |
|
|
import asyncio |
|
|
from contextlib import asynccontextmanager |
|
|
|
|
|
@asynccontextmanager |
|
|
async def lifespan(app: FastAPI): |
|
|
global needModel |
|
|
needModel = pipeline("sentiment-analysis") |
|
|
yield |
|
|
|
|
|
app = FastAPI(lifespan=lifespan) |
|
|
|
|
|
class Request(BaseModel): |
|
|
text: str |
|
|
number: int |
|
|
|
|
|
@app.post("/message") |
|
|
async def model(request: Request): |
|
|
try: |
|
|
|
|
|
return {"result": request.text.upper()} |
|
|
except Exception as e: |
|
|
return {"error": str(e)} |
|
|
|