Update app.py

app.py

@@ -16,10 +16,8 @@ def preprocess_image(image):
 def predict_pneumonia(image):
     img_array = preprocess_image(image)
     prediction = model.predict(img_array)[0][0]
-    pneumonia_percent = np.clip(prediction * 100, 0, 100)
-    normal_percent = np.clip((1 - prediction) * 100, 0, 100)
-    return {"Pneumonia": pneumonia_percent, "Normal": normal_percent}
-
+    pneumonia_prediction = 1 if prediction >= 0.5 else 0
+    return {"Pneumonia": pneumonia_prediction}
 
 inputs = gr.inputs.Image(shape=(180, 180))
 outputs = gr.outputs.Label(num_top_classes=2)