Update backend/app.py

backend/app.py

@@ -217,13 +217,14 @@ async def predict(model_name: str = Form(...), file: UploadFile = File(...)):
         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))}
+        predicted_confidence = float(proba[pred])
         return {
             "model_used": "CIN Classifier",
             "prediction": predicted_label,
-            "confidence": confidences
+            "confidence": predicted_confidence
         }
 
+
     elif model_name == "histopathology":
         result = predict_histopathology(image)
         return result