malavikapradeep2001 commited on
Commit
fb6143f
·
unverified ·
1 Parent(s): 3bf45a4

Update backend/app.py

Browse files
Files changed (1) hide show
  1. backend/app.py +1 -4
backend/app.py CHANGED
@@ -160,10 +160,7 @@ def predict_histopathology(image: Image.Image):
160
  return {
161
  "model_used": "Breast Cancer Histopathology Classifier",
162
  "prediction": class_names[predicted_class],
163
- "confidence": float(np.max(prediction_proba)),
164
- "probabilities": {
165
- "Benign": float(prediction_proba[0]),
166
- "Malignant": float(prediction_proba[1])
167
  }
168
  }
169
 
 
160
  return {
161
  "model_used": "Breast Cancer Histopathology Classifier",
162
  "prediction": class_names[predicted_class],
163
+ "confidence": float(np.max(prediction_proba))
 
 
 
164
  }
165
  }
166