diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 3568876..e3de6d2 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -19,7 +19,7 @@ jobs:
 
     steps:
       - name: Checkout
-        uses: actions/checkout@v4
+        uses: actions/checkout@v6
 
       - name: Setup Python
         uses: actions/setup-python@v5
diff --git a/.github/workflows/push.yml b/.github/workflows/push.yml
index 81eb2ee..74d87dd 100644
--- a/.github/workflows/push.yml
+++ b/.github/workflows/push.yml
@@ -18,7 +18,7 @@ jobs:
         model: [yolo11n, yolov8s-worldv2, yoloe11s]
     steps:
       - name: Checkout
-        uses: actions/checkout@v4
+        uses: actions/checkout@v6
       - name: Setup Python
         uses: actions/setup-python@v5
         with:
diff --git a/yolo11n/predict.py b/yolo11n/predict.py
index ad27f4d..3dc368c 100644
--- a/yolo11n/predict.py
+++ b/yolo11n/predict.py
@@ -1,6 +1,6 @@
 # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
 
-from typing import Optional
+from __future__ import annotations
 
 from cog import BaseModel, BasePredictor, Input, Path
 from ultralytics import YOLO
@@ -9,8 +9,8 @@
 class Output(BaseModel):
     """Output model for predictions."""
 
-    image: Optional[Path] = None
-    json_str: Optional[str] = None
+    image: Path | None = None
+    json_str: str | None = None
 
 
 class Predictor(BasePredictor):
diff --git a/yoloe11s/predict.py b/yoloe11s/predict.py
index 85c2532..94c8dd3 100644
--- a/yoloe11s/predict.py
+++ b/yoloe11s/predict.py
@@ -1,6 +1,6 @@
 # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
 
-from typing import Optional
+from __future__ import annotations
 
 from cog import BaseModel, BasePredictor, Input, Path
 from ultralytics import YOLOE
@@ -9,8 +9,8 @@
 class Output(BaseModel):
     """Output model for predictions."""
 
-    image: Optional[Path] = None
-    json_str: Optional[str] = None
+    image: Path | None = None
+    json_str: str | None = None
 
 
 class Predictor(BasePredictor):
diff --git a/yolov8s-worldv2/predict.py b/yolov8s-worldv2/predict.py
index 3288bb8..7933193 100644
--- a/yolov8s-worldv2/predict.py
+++ b/yolov8s-worldv2/predict.py
@@ -1,6 +1,6 @@
 # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
 
-from typing import Optional
+from __future__ import annotations
 
 from cog import BaseModel, BasePredictor, Input, Path
 from ultralytics import YOLOWorld
@@ -9,8 +9,8 @@
 class Output(BaseModel):
     """Output model for predictions."""
 
-    image: Optional[Path] = None
-    json_str: Optional[str] = None
+    image: Path | None = None
+    json_str: str | None = None
 
 
 class Predictor(BasePredictor):