Update app.py

app.py

@@ -22,24 +22,9 @@ def predict_pneumonia(image):
     threshold = 0.5
 
     if prediction >= threshold:
-        pneumonia_prediction = 1
-        class_name = "Pneumonia"
-        pneumonia_probability = prediction
-        normal_probability = 1 - prediction
+        return "Pneumonia"
     else:
-        pneumonia_prediction = 0
-        class_name = "Normal"
-        pneumonia_probability = 1 - prediction
-        normal_probability = prediction
-
-    return {
-        "Pneumonia Prediction": pneumonia_prediction,
-        "Class Name": class_name,
-        "Pneumonia Probability": pneumonia_probability,
-        "Normal Probability": normal_probability
-    }
-
-
+        return "Normal"
 
 inputs = gr.inputs.Image(shape=(180, 180))
 outputs = gr.outputs.Label(num_top_classes=2)