From ff4f99ecf40d2f48c7071b51bbce2d86e1bd37cd Mon Sep 17 00:00:00 2001
From: Bingsu <ks2515@naver.com>
Date: Wed, 26 Apr 2023 23:08:30 +0900
Subject: [PATCH] fix: boxes attribute error

---
 adetailer/ultralytics.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/adetailer/ultralytics.py b/adetailer/ultralytics.py
index 6902172..927de7a 100644
--- a/adetailer/ultralytics.py
+++ b/adetailer/ultralytics.py
@@ -18,7 +18,7 @@ def ultralytics_predict(
     model = YOLO(model_path)
     pred = model(image, conf=confidence, hide_labels=True)
 
-    bboxes = pred[0].xyxy.cpu().numpy()
+    bboxes = pred[0].boxes.xyxy.cpu().numpy()
     if bboxes.size == 0:
         return PredictOutput()
     bboxes = bboxes.tolist()