slake / app.py
Salcidio
curl
6040afc
Raw
History Blame Contribute Delete
328 Bytes
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