Update app.py

app.py

@@ -38,18 +38,19 @@ def predict(image_path):
     predicted_class = 'Pneumonia' if predicted_probabilities > 0.5 else 'Normal' 
     confidence_score = predicted_probabilities if predicted_class == 'Pneumonia' else 1 - predicted_probabilities
     confidence_score *= 100
-    confidence_score = str(confidence_score) + "%"
 
-    return confidence_score, predicted_class
+    return predicted_class, confidence_score
 
 image_component = gr.Image(type="filepath", label="Chest X-ray Image")
 label_component = gr.Label(label="Classification result", num_top_classes=1)
+confidence_component = gr.Label(label="Confidence Score (%)")
+
 description = ""
 title = "Pneumonia Classifier"
 
 demo = gr.Interface(fn=predict, 
                     inputs=image_component, 
-                    outputs=label_component, 
+                    outputs=[label_component, confidence_component], 
                     description=description, 
                     title=title,
                     examples=["demo_example_imgs/NORMAL-0001-0001.jpeg",