🐛 Fix pipeline

MultimediaTechLab · Feb 23, 2025 · 3bfa804 · 3bfa804
1 parent 89ea875
commit 3bfa804
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yolo/utils/model_utils.py b/yolo/utils/model_utils.py
@@ -179,7 +179,7 @@ def __call__(
         else:
             prediction = predict
 
-        pred_class, _, pred_bbox = predict[:3]
+        pred_class, _, pred_bbox = prediction[:3]
         pred_conf = prediction[3] if len(prediction) == 4 else None
 
         if rev_tensor is not None: