Update app.py

app.py

@@ -20,7 +20,7 @@ logger = logging.getLogger(__name__)
 
 
 # 下載YOLOv8預訓練模型
-model_yolo = YOLO('yolov8s.pt')  # 使用 YOLOv8 預訓練模型
+model_yolo = YOLO('yolov8n.pt')  # 使用 YOLOv8 預訓練模型
 
 
 dog_breeds = ["Afghan_Hound", "African_Hunting_Dog", "Airedale", "American_Staffordshire_Terrier", 
@@ -167,7 +167,7 @@ async def predict_single_dog(image):
     return top1_prob, topk_breeds, topk_probs_percent
     
 
-async def detect_multiple_dogs(image, conf_threshold=0.4, iou_threshold=0.5):
+async def detect_multiple_dogs(image, conf_threshold=0.4, iou_threshold=0.55):
     results = model_yolo(image, conf=conf_threshold, iou=iou_threshold)[0]
     dogs = []
     boxes = []