Update app.py

app.py

@@ -60,16 +60,12 @@ async def predict_endpoint(
     
     response = {
         "prediction": "epilepsy" if pred_label == 1 else "no epilepsy",
-        "confidence": mean_prob
+        "mean_probability": float(mean_prob),
+        "segment_probabilities": [float(p) for p in segment_probs]
     }
+
     print("Prediction complete, returning 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)
+    return JSONResponse(content=response)
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)