changed type of boxes

utils.py

@@ -10,14 +10,14 @@ from tqdm import tqdm
 def plot_predictions(
         image: str,
         labels: list[str],
-        scores: torch.Tensor,
-        boxes: torch.Tensor,
+        scores: list[float],
+        boxes: list[float],
     ) -> np.ndarray:
 
     image_source = cv2.imread(image)
     image_source = cv2.cvtColor(image_source, cv2.COLOR_BGR2RGB)
 
-    boxes = sv.Detections(xyxy=boxes.cpu().numpy())
+    boxes = sv.Detections(xyxy=boxes)
 
     labels = [
         f"{phrase} {logit:.2f}"