Update detect.py

detect.py

@@ -100,7 +100,7 @@ def run(
     
         # NMS
         with dt[2]:
-            # pred = pred[0][1] if isinstance(pred[0], list) else pred[0]  # single model or ensemble
+            pred = pred[0][1] if isinstance(pred[0], list) else pred[0]  # single model or ensemble
             pred = non_max_suppression(pred, conf_thres, iou_thres, classes, agnostic_nms, max_det=max_det)