diff --git a/backend/Dockerfile b/backend/Dockerfile index 734a7cb..3706205 100644 --- a/backend/Dockerfile +++ b/backend/Dockerfile @@ -16,7 +16,8 @@ RUN apt-get update \ COPY requirements.txt ./requirements.txt # Install CPU-only PyTorch wheels so the container does not pull CUDA runtimes. -RUN pip install --extra-index-url https://download.pytorch.org/whl/cpu -r requirements.txt +RUN pip install --extra-index-url https://download.pytorch.org/whl/cpu -r requirements.txt \ + && pip install --no-deps facenet-pytorch==2.6.0 COPY . . diff --git a/backend/requirements.txt b/backend/requirements.txt index 4be2aa8..f4d937b 100644 --- a/backend/requirements.txt +++ b/backend/requirements.txt @@ -4,10 +4,9 @@ python-multipart==0.0.20 email-validator==2.3.0 firebase-admin==7.1.0 transformers==4.57.1 -torch==2.2.2+cpu -torchvision==0.17.2+cpu +torch==2.6.0+cpu +torchvision==0.21.0+cpu opencv-python-headless==4.11.0.86 -facenet-pytorch==2.6.0 timm==1.0.19 Pillow==10.2.0 numpy==1.26.4