Update main.py

main.py

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