Update app.py

app.py

@@ -25,7 +25,17 @@ def predict_pneumonia(image):
     else:
         pneumonia_prediction = 0
 
-    return {"Pneumonia": pneumonia_prediction}
+    # Return the probability of each class
+    class_probabilities = model.predict(img_array)
+
+    # Return the top two possible classifications
+    top_classes = model.predict_classes(img_array)
+
+    return {
+        "Pneumonia": pneumonia_prediction,
+        "Class probabilities": class_probabilities,
+        "Top classes": top_classes
+    }
 
 
 inputs = gr.inputs.Image(shape=(180, 180))