Update app.py

app.py

@@ -17,13 +17,12 @@ def predict(image):
     # Get the predicted class label
     if prediction[0][0] < 0.5:
         label = 'Benign'
+        confidence = 1 - prediction[0][0]  # Confidence for benign
     else:
         label = 'Malignant'
+        confidence = prediction[0][0]  # Confidence for malignant
     
-    # Get the sigmoid percentage
-    sigmoid_percentage = f"{prediction[0][0] * 100:.2f}%"
-    
-    return f"{label} ({sigmoid_percentage})", sigmoid_percentage
+    return {'label': label, 'confidence': confidence}
 
 examples = [["benign.jpg"], ["malignant.jpg"]]