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

@app.post("/summarize")
async def summarize(item: TextIn):
    response = model(item.text)
    return response