Update app.py

app.py

@@ -24,21 +24,23 @@ def predict_pneumonia(image):
     if prediction >= threshold:
         pneumonia_prediction = 1
         class_name = "Pneumonia"
+        pneumonia_probability = prediction
+        normal_probability = 1 - prediction
     else:
         pneumonia_prediction = 0
         class_name = "Normal"
-
-    # Return the probability of each class
-    class_probabilities = model.predict(img_array)[0]
+        pneumonia_probability = 1 - prediction
+        normal_probability = prediction
 
     return {
         "Pneumonia Prediction": pneumonia_prediction,
         "Class Name": class_name,
-        "Pneumonia Probability": class_probabilities[0],
-        "Normal Probability": class_probabilities[1]
+        "Pneumonia Probability": pneumonia_probability,
+        "Normal Probability": normal_probability
     }
 
 
+
 inputs = gr.inputs.Image(shape=(180, 180))
 outputs = gr.outputs.Label(num_top_classes=2)