Update app.py

app.py

@@ -1,22 +1,19 @@
 import gradio as gr
 from PIL import Image
 from ultralytics import YOLO
-import numpy as np
 
-# Load your model
+# Load the model with original name
 model = YOLO("best (2).pt")
 
-# Prediction function
 def detect_teeth(image):
-    results = model.predict(image, conf=0.26)
+    results = model.predict(image)
 
-    # Optional post-processing
     r = results[0]
     if r.masks is not None:
         r.masks.data = (r.masks.data > 0.3).float()
 
-    # Create segmented image with bounding boxes and masks
-    output_img = r.plot()
+    # Render without confidence scores or labels
+    output_img = r.plot(conf=False, labels=False, boxes=True)
     return Image.fromarray(output_img)
 
 # Gradio interface
@@ -24,8 +21,8 @@ interface = gr.Interface(
     fn=detect_teeth,
     inputs=gr.Image(type="pil", label="Upload Dental X-ray"),
     outputs=gr.Image(type="pil", label="Detection Result"),
-    title="Dental X-ray Detection",
-    description="Upload a dental X-ray to see segmentation and detection results using YOLOv8."
+    title="Dental Segmentation App",
+    description="Upload a dental X-ray to see the segmentation result using YOLOv8."
 )
 
 interface.launch()