Update app.py

app.py

@@ -22,19 +22,17 @@ def predict(image):
         label = 'Malignant'
         confidence = prediction[0][0]  # Confidence for malignant
     
-    return {'label': label, 'confidence': float(confidence)}  # Convert confidence to float
+    return label, confidence
 
-# Custom post-processing function to format the label and confidence
-def custom_postprocess(output):
-    label = output['label']
-    confidence = output['confidence'] * 100
-    return f"{label} ({confidence:.2f}%)"
+# Custom component to display label with confidence
+def label_with_confidence(label, confidence):
+    return f"{label} ({confidence * 100:.2f}%)"
 
 examples = [["benign.jpg"], ["malignant.jpg"]]
 
-# Define input and output components
+# Define input and custom output components
 image_input = gr.inputs.Image(shape=(150, 150))
-label_output = gr.outputs.Label()
+label_output = gr.outputs.Component(label_with_confidence, inputs=["label", "confidence"])
 
 # Define a Gradio interface for user interaction
 iface = gr.Interface(