Update app.py

app.py

@@ -63,7 +63,13 @@ async def predict_endpoint(
         "confidence": mean_prob
     }
     print("Prediction complete, returning response...")
-    return JSONResponse(content=response)
+    # Conversion des types numpy → Python natifs
+    response = {
+    "prediction": str(final_label),
+    "mean_probability": float(mean_prob),  # <-- ✅ ici
+    "segment_probabilities": [float(p) for p in segment_probs]  # <-- ✅ ici
+}
+return JSONResponse(content=response)
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)