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


class InputText(BaseModel):
    text: str


app = FastAPI()

@app.get("/")
def greet_json():
    return {"Hello": "World!"}


@app.get("/summary")
def summary(inp: InputText):
    return summarizer(inp.text, max_length=1000, min_length=30, do_sample=False)