Update Dockerfile

Dockerfile

@@ -12,7 +12,12 @@ RUN useradd -m -u 1000 appuser
 WORKDIR /app
 
 COPY requirements.txt .
-RUN pip install --no-cache-dir --upgrade "pip==24.3.1" && \
+
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir \
+        torch==2.2.2+cpu \
+        torchvision==0.17.2+cpu \
+        --extra-index-url https://download.pytorch.org/whl/cpu && \
     pip install --no-cache-dir -r requirements.txt
 
 COPY . .