diff --git a/personfromvid/models/head_pose_estimator.py b/personfromvid/models/head_pose_estimator.py
index 818cf86..435bc38 100644
--- a/personfromvid/models/head_pose_estimator.py
+++ b/personfromvid/models/head_pose_estimator.py
@@ -140,7 +140,7 @@ def __init__(
 
     def _resolve_device(self, device: str) -> str:
         """Resolve device string to actual device."""
-        if device == "auto":
+        if device in ["auto", "gpu"]:
             # Check for CUDA availability
             try:
                 import torch