Spaces:
Sleeping
Sleeping
File size: 575 Bytes
d24bd61 eb90c52 2d92851 eb90c52 6915dbd d24bd61 eb90c52 9f761aa eb90c52 020dc7e 6365a36 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
from fastapi import FastAPI
from pydantic import BaseModel
from typing import Optional
from transformers import pipeline
app = FastAPI()
summarizer = pipeline(
"summarization",
# model="Falconsai/text_summarization"
model="pszemraj/long-t5-tglobal-base-16384-book-summary"
)
class InputText(BaseModel):
text: str
app = FastAPI()
@app.get("/")
def greet_json():
return {"Hello": "World!"}
@app.post("/summary")
def summary(inp: InputText):
print(inp.text)
return summarizer(inp.text[:3600], max_length=2048, min_length=30, do_sample=False) |