Use Torch CPU in our docker builds (#191)

* Explicitly use torch cpu

Author: haplesshero13

.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

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 \