Update app.py

app.py

@@ -50,15 +50,14 @@ def diagnose_image(image, model, input_shape):
     result_text = f"Probability of glaucoma: {glaucoma_probability:.2%}"
     
     img_display = np.array(image)
-    if img_display.shape[2] == 1:  # Convert to RGB for display
-        img_display = cv2.cvtColor(img_display.squeeze(), cv2.COLOR_GRAY2RGB)
+    if len(img_display.shape) == 2 or img_display.shape[2] == 1:  # Convert to RGB for display
+        img_display = cv2.cvtColor(img_display, cv2.COLOR_GRAY2RGB)
     image_pil = Image.fromarray(img_display)
     draw = ImageDraw.Draw(image_pil)
     font = ImageFont.load_default()
     
     text = f"{glaucoma_probability:.2%}"
-    text_bbox = draw.textbbox((0, 0), text, font=font)
-    text_size = (text_bbox[2] - text_bbox[0], text_bbox[3] - text_bbox[1])
+    text_size = draw.textsize(text, font=font)
     
     rect_width = 200
     rect_height = 100
@@ -77,7 +76,7 @@ def diagnose_image(image, model, input_shape):
 def main():
     with gr.Blocks() as demo:
         gr.Markdown("# Glaucoma Detection App")
-        gr.Markdown("Upload an fundus eye image to detect the probability of glaucoma.")
+        gr.Markdown("Upload a fundus eye image to detect the probability of glaucoma.")
         
         with gr.Row():
             model_file = gr.File(label="Upload Model (.h5 or .keras)")
@@ -98,7 +97,8 @@ def main():
         input_shape = gr.State(None)
         
         def diagnose_and_display(image, model, input_shape):
-            return diagnose_image(image, model, input_shape)
+            diagnosis_image, result_text = diagnose_image(image, model, input_shape)
+            return diagnosis_image, result_text
         
         load_model_btn.click(fn=load_and_display_model_info, inputs=model_file, outputs=[model, model_info, input_shape])
         submit_btn.click(fn=diagnose_and_display, inputs=[image, model, input_shape], outputs=[image, result])