Update app.py

app.py

@@ -41,7 +41,7 @@ def detect_gingivitis(image, conf_threshold=0.5, iou_threshold=0.5):
     # For now, return it for integration
     
     # ✅ Return in same order as outputs 
-    return img_swelling, img_redness, img_bleeding, has_gingivitis
+    return img_swelling, img_redness, img_bleeding, "Gingivitis Detected" if has_gingivitis else "No Gingivitis Detected"
 
 
 def preprocess_image(image, max_size=1024, contrast_factor=1.05):
@@ -99,7 +99,7 @@ interface = gr.Interface(
         gr.Image(label="Swelling Detection"), 
         gr.Image(label="Redness Detection"), 
         gr.Image(label="Bleeding Detection"), 
-        gr.Label(label="Gingivitis Detected?")
+        gr.Label(label="Gingivitis Diagnosis")
     ], 
     title="Gingivitis Detection (3 Models)", 
     description="Detect swelling, redness, and bleeding using 3 YOLO models.", )