diff --git a/detect.py b/detect.py
index cfd91665..61d7969d 100755
--- a/detect.py
+++ b/detect.py
@@ -67,7 +67,7 @@ def detect(
         # Get detections
         with torch.no_grad():
             img = torch.from_numpy(img).unsqueeze(0).to(device)
-            # pred = torch.onnx._export(model, img, 'weights/model.onnx', verbose=True,); return  # ONNX export
+            # pred = torch.onnx._export(model, img, 'weights/model.onnx', verbose=True); return  # ONNX export
             pred = model(img)
             pred = pred[pred[:, :, 4] > conf_thres]