Update app.py

app.py

@@ -27,8 +27,6 @@ def detect_periodontitis(image_np):
 
     # Convert OpenCV BGR → RGB for Gradio display
     combined_rgb = cv2.cvtColor(results["combined"], cv2.COLOR_BGR2RGB)
-    cej_mask_rgb = cv2.cvtColor(results["cej_mask"] * 255, cv2.COLOR_GRAY2RGB)
-    abc_mask_rgb = cv2.cvtColor(results["abc_mask"] * 255, cv2.COLOR_GRAY2RGB)
 
     # Optional: summarize measurements for text output
     summaries = []
@@ -43,7 +41,7 @@ def detect_periodontitis(image_np):
 
     summary_text = "\n".join(summaries) if summaries else "No detections found."
 
-    return combined_rgb, cej_mask_rgb, abc_mask_rgb, summary_text
+    return combined_rgb, summary_text
 
 
 # ==========================
@@ -54,8 +52,6 @@ demo = gr.Interface(
     inputs=gr.Image(type="numpy", label="Upload Dental X-Ray"),
     outputs=[
         gr.Image(label="Final Annotated Image (YOLO + CEJ–ABC)"),
-        gr.Image(label="CEJ Segmentation Mask"),
-        gr.Image(label="ABC Segmentation Mask"),
         gr.Textbox(label="Analysis Summary"),
     ],
     title="🦷 Periodontitis Detection & Analysis",