Update app.py

app.py

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