| from fastapi import FastAPI | |
| from pydantic import BaseModel | |
| from transformers import pipeline | |
| app = FastAPI() | |
| model = pipeline("summarization", model="facebook/bart-large-cnn") | |
| class TextIn(BaseModel): | |
| text: str | |
| async def summarize(item: TextIn): | |
| response = model(item.text) | |
| return response |