Update app.py

app.py

@@ -16,9 +16,15 @@ def preprocess_image(image):
 def predict_pneumonia(image):
     img_array = preprocess_image(image)
     prediction = model.predict(img_array)[0][0]
-    pneumonia_prediction = 1 if prediction >= 0.5 else 0
+
+    if prediction >= 0.5:
+        pneumonia_prediction = 1
+    else:
+        pneumonia_prediction = 0
+
     return {"Pneumonia": pneumonia_prediction}
 
+
 inputs = gr.inputs.Image(shape=(180, 180))
 outputs = gr.outputs.Label(num_top_classes=2)