Update Dockerfile

Dockerfile

@@ -11,14 +11,19 @@ RUN apt-get update && apt-get install -y \
 RUN useradd -m -u 1000 user
 USER user
 ENV PATH="/home/user/.local/bin:$PATH"
-
 WORKDIR /app
 
 COPY --chown=user requirements.txt .
+
+# ── Install PyTorch CPU FIRST, explicitly, before everything else ──
+RUN pip install --no-cache-dir \
+    torch==2.3.0+cpu \
+    torchvision==0.18.0+cpu \
+    --index-url https://download.pytorch.org/whl/cpu
+
+# ── Then install the rest ──
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 COPY --chown=user . .
-
 EXPOSE 7860
-
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]