Spaces:
Sleeping
Sleeping
Update backend/app.py
Browse files- 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 |
|