Upload app.py

app.py

@@ -73,10 +73,21 @@ from fastapi import Request
 @app.post("/ner")
 @limiter.limit("10/minute")
 def ner_predict(body: TextRequest, request: Request, api_key: str = Depends(verify_token)):
-    logger.info("NER request from IP: %s", get_remote_address(request))
-    predictions = ner_pipeline(body.text)
+    logger.info("Received NER request from IP: %s", get_remote_address(request))
+    raw_predictions = ner_pipeline(body.text)
+
+    # Convert all numpy.float32 scores to float
+    predictions = [
+        {
+            **ent,
+            "score": float(ent["score"])
+        }
+        for ent in raw_predictions
+    ]
+
     API_KEYS[api_key]["usage_count"] = API_KEYS[api_key].get("usage_count", 0) + 1
     save_keys()
+
     return {
         "entities": predictions,
         "usage": API_KEYS[api_key]["usage_count"]