Update app.py

app.py

@@ -2,6 +2,8 @@ from fastapi import FastAPI
 from fastapi.responses import JSONResponse
 from transformers import pipeline
 from fastapi.requests import Request
+from pydantic import BaseModel
+
 
 app = FastAPI()
 
@@ -20,13 +22,18 @@ def read_root():
         "message": "Welcome to the FastAPI text generation API! Use the /generate endpoint with a 'text' query parameter."
     }
 
-@app.get("/generate")
-def generate(text: str):
+# Define a request model for the POST payload
+class GenerateRequest(BaseModel):
+    text: str
+
+@app.post("/generate")
+def generate_post(request: GenerateRequest):
     try:
-        output = pipe(text)
+        output = pipe(request.text)
         return {"output": output[0]["generated_text"]}
     except Exception as e:
         return JSONResponse(
             status_code=500,
             content={"error": f"An error occurred: {str(e)}"}
         )
+