diff --git a/.github/workflows/build_images.yml b/.github/workflows/build_images.yml
index aaf13e56..6d57d9f5 100755
--- a/.github/workflows/build_images.yml
+++ b/.github/workflows/build_images.yml
@@ -53,7 +53,7 @@ jobs:
           tags: ${{ steps.base.outputs.tags }}
           cache-from: type=registry,ref=${{ steps.login-ecr.outputs.registry }}/dev/ndif_base:cache
           cache-to: type=registry,ref=${{ steps.login-ecr.outputs.registry }}/dev/ndif_base:cache,mode=max
-      
+
       - name: Tag the API image
         id: api
         uses: docker/metadata-action@v5
diff --git a/docker/Dockerfile b/docker/Dockerfile
index 066bc01d..cc131a77 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -11,6 +11,7 @@ RUN apt-get update -y \
 
 # Copy and install base requirements
 COPY src/services/base/requirements.in /tmp/requirements.in
+RUN uv pip install --system torch --index-url https://download.pytorch.org/whl/cpu
 RUN uv pip install --system -r /tmp/requirements.in \
     && find /usr/local/lib/python3.12 -type d -name "__pycache__" -exec rm -rf {} + 2>/dev/null || true \
     && find /usr/local/lib/python3.12 -name "*.pyc" -delete \