Update app.py

app.py

@@ -171,7 +171,7 @@ async def detect_multiple_dogs(image, conf_threshold=0.35, iou_threshold=0.55):
     dogs = []
     boxes = []
     for box in results.boxes:
-        if box.cls == 16:  # COCO dataset class for dog is 16
+        if box.cls == 16:  # COCO dataset class for dog is 16 
             xyxy = box.xyxy[0].tolist()
             confidence = box.conf.item()
             boxes.append((xyxy, confidence))
@@ -287,7 +287,7 @@ async def predict(image):
             top1_prob, topk_breeds, topk_probs_percent = await predict_single_dog(cropped_image)
             color = color_list[i % len(color_list)]
             draw.rectangle(box, outline=color, width=3)
-            draw.text((box[0] + 5, box[1] + 5), f"Dog {i+1}", fill=color, font=font)
+            draw.text((box[0] + 5, box[1] + 5), f"Dog {i+1}", fill=color, font=font) # Adjust the mark place
 
 
             combined_confidence = detection_confidence * top1_prob