| 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)} |
|
|