Update app.py

app.py

@@ -4,6 +4,8 @@ from starlette.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 import uvicorn
 import torch
+from fastapi.encoders import jsonable_encoder
+from fastapi.responses import JSONResponse
 from simpletransformers.classification import ClassificationModel
 app = FastAPI()
 
@@ -27,8 +29,10 @@ async def c(query:Query):
     try:
         predictions, raw_outputs = rob_chem_model.predict([str(query.query)])
         print(predictions)
-        answer = {"prediction":predictions[0]}
-        return answer
+        json_compatible_item_data = jsonable_encoder(predictions[0])
+        return JSONResponse(content=json_compatible_item_data)
+        #answer = {"prediction":predictions[0]}
+        #return answer
         
 
     except Exception as e: