curl

app.py

@@ -1,5 +1,14 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
 from transformers import pipeline
 
-model = pipeline("summarization", model= "facebook/bart-large-cnn")
-response = model("Text to summarize")
-print(response)
+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