Update backend/app.py

backend/app.py

@@ -22,7 +22,7 @@ if hf_token:
 
 from fastapi import FastAPI, File, UploadFile, Form
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse
+from fastapi.responses import JSONResponse, FileResponse
 from fastapi.staticfiles import StaticFiles
 from ultralytics import YOLO
 from io import BytesIO
@@ -216,10 +216,12 @@ async def predict(model_name: str = Form(...), file: UploadFile = File(...)):
         pred = clf.predict(X_scaled)[0]
         proba = clf.predict_proba(X_scaled)[0]
         classes = ["CIN1", "CIN2", "CIN3"]
+        predicted_label = classes[pred]
+        confidences = {classes[i]: float(proba[i]) for i in range(len(classes))}
         return {
             "model_used": "CIN Classifier",
-            "prediction": classes[pred],
-            "probabilities": dict(zip(classes, map(float, proba)))
+            "prediction": predicted_label,
+            "confidence": confidences
         }
 
     elif model_name == "histopathology":