json error fix

main.py

@@ -4,6 +4,8 @@ import sys
 import os
 from fastapi import FastAPI
 from model_utils import load_model_and_weights, single_predict
+import json
+
 
 app = FastAPI()
 
@@ -23,12 +25,12 @@ async def root():
                  "happens on this show every 0 seconds. I don't think that Days of our lives is a good show, but I seem to be addicted to it "
                  "anyway. PROPNAME is so nice and her and LOCNAME are finally together, but probably not for long because there is")
     predictions = single_predict(model, test_text)
-    return {"predictions": predictions}
+    return JSONResponse(content=predictions)
 
 @app.get("/predict")
 async def predict_personality_get(text: str):
     predictions = single_predict(model, text)
-    return {"predictions": predictions}
+    return JSONResponse(content=predictions)
 
 if __name__ == "__main__":
     import uvicorn

model_utils.py

@@ -74,24 +74,17 @@ def tokenize_text(text, hugging_model='roberta-base'):
     return x
 
 def single_predict(model, text, traits=['cAGR', 'cCON', 'cEXT', 'cOPN', 'cNEU']):
-    print("predict function-----")
     traits_scores = dict()
     predicted_labels = dict()
     x = tokenize_text([text])
     logits = model.predict(x, verbose=0).logits
-    print("logits function-----")
     probs = tf.math.sigmoid(logits).numpy()
-    print("sigmoid function-----")
     predictions = np.where(probs > 0.5, 1, 0)
-    print("predictions function------")
     for t, s in zip(traits, probs[0]):
-        traits_scores[t] = s
-    print("t, s in")
+        traits_scores[t] = float(s)  # Convert numpy.float32 to Python float
     for t, l in zip(traits, predictions[0]):
-        predicted_labels[t] = l
-    print("t, l in")
+        predicted_labels[t] = int(l)  # Convert numpy.int64 to Python int
     final_dic = {'probability': traits_scores, 'predicted_label': predicted_labels}
-    print("end predict function------")
     return final_dic
 
 def load_model_and_weights(hugging_model='roberta-base', output_folder='.'):