Update Dockerfile

Dockerfile

@@ -13,8 +13,15 @@ WORKDIR /app
 
 # ---- Python dependencies ----
 COPY requirements.txt /app/requirements.txt
-RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir -r /app/requirements.txt
+RUN pip install --no-cache-dir --upgrade pip
+
+# Install torch from the CPU-only wheel index first. This avoids pulling
+# in the ~2GB of NVIDIA CUDA/cuDNN packages that the default PyPI wheel
+# depends on, which are useless (and wasteful) on this CPU-only Space.
+RUN pip install --no-cache-dir torch==2.3.1 --index-url https://download.pytorch.org/whl/cpu
+
+# Install the remaining Python dependencies.
+RUN pip install --no-cache-dir -r /app/requirements.txt
 
 # ---- Pre-download the model at build time ----
 # This bakes the model weights into the image so the container starts