Update Dockerfile

Dockerfile

@@ -1,20 +1,20 @@
 FROM python:3.10-slim
 
-# Set working directory
 WORKDIR /app
 
-# System dependencies
 RUN apt-get update && apt-get install -y \
     libgl1 \
     libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy files
 COPY requirements.txt ./
+
+# Install torch & torchvision from PyTorch CPU wheels
+RUN pip install --no-cache-dir torch==2.0.1+cpu torchvision==0.15.2+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
+
 RUN pip install --no-cache-dir -r requirements.txt
 
 COPY app.py .
 
 EXPOSE 5000
-
 CMD ["python", "app.py"]