Update Dockerfile

Dockerfile

@@ -11,14 +11,14 @@ RUN useradd -m -u 1000 appuser
 
 WORKDIR /app
 
-COPY requirements.txt .
-
+# Download and install torch CPU wheel directly
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir \
-        torch==2.2.2+cpu \
-        torchvision==0.17.2+cpu \
-        --index-url https://download.pytorch.org/whl/cpu
+        "torch==2.2.2" \
+        "torchvision==0.17.2" \
+        --find-links https://download.pytorch.org/whl/cpu/torch_stable.html
 
+COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
 COPY . .