Update app.py

app.py

@@ -14,25 +14,22 @@ def predict(image):
     # Make a prediction using the model
     prediction = model.predict(image)
     
+    # Get the sigmoid percentage
+    sigmoid_percentage = prediction[0][0] * 100
+    
     # 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
     
-    return label, confidence
-
-# Custom component to display label with confidence
-def label_with_confidence(label, confidence):
-    return f"{label} ({confidence * 100:.2f}%)"
+    return f"{label} ({sigmoid_percentage:.2f}%)"
 
 examples = [["benign.jpg"], ["malignant.jpg"]]
 
-# Define input and custom output components
+# Define input and output components
 image_input = gr.inputs.Image(shape=(150, 150))
-label_output = gr.outputs.Component(label_with_confidence, inputs=["label", "confidence"])
+label_output = gr.outputs.Label()
 
 # Define a Gradio interface for user interaction
 iface = gr.Interface(