Update app.py

app.py

@@ -1,7 +1,23 @@
 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)