Update main.py

main.py

@@ -1,4 +1,5 @@
 from fastapi import FastAPI
+from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 from transformers import pipeline
 import uvicorn
@@ -18,5 +19,6 @@ async def classify_text(text):
 
 @app.post("/classify/")
 async def classify(payload: Payload):
-    return {"response": classify_text(payload.text)}
+    result = await classify_text(payload.text)
+    return JSONResponse({"data": result})