Update app.py

app.py

@@ -9,17 +9,19 @@ model = YOLO('Dental_model.pt')  # Replace 'yolov8n.pt' with your model file if
 pic_files=glob.glob('*.jpg')
 names=model.model.names
 def detect_objects(image):
-    image1=image.copy()
+    image1 = image.copy()
     results = model.predict(image)
-    whole=results[0].plot()
-    classes=results[0].boxes.cls.cpu().tolist()
-    boxes=results[0].boxes.xyxy.cpu()
+    classes = results[0].boxes.cls.cpu().tolist()
+    boxes = results[0].boxes.xyxy.cpu()
     annotator = Annotator(image, line_width=3)
-    annotator1=Annotator(image1, line_width=3)
-    for box,cls in zip(boxes,classes):
+    annotator1 = Annotator(image1, line_width=3)
+    
+    for box, cls in zip(boxes, classes):
         annotator.box_label(box, label=names[int(cls)], color=colors(int(cls)))
         annotator1.box_label(box, label=None, color=colors(int(cls)))
-    return Image.fromarray(annotator.result()),Image.fromarray(annotator1.result()),None
+
+    return Image.fromarray(annotator.result()), Image.fromarray(annotator1.result())
+
 
 # Gradio Interface
 title = "YOLOv8 Object Detection"