Update Dockerfile

Dockerfile

@@ -2,25 +2,27 @@ FROM python:3.10
 
 WORKDIR /app
 
-# System dependencies
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
-    libgl1 \
-    libglib2.0-0 \
-    && rm -rf /var/lib/apt/lists/*
+    libgl1 libglib2.0-0 libsm6 libxext6 libxrender-dev && \
+    rm -rf /var/lib/apt/lists/*
 
-# Ensure pip is up to date
+# Upgrade pip
 RUN pip install --upgrade pip
 
-# Install numpy first, then torch/torchvision (CPU)
-RUN pip install numpy
+# Install NumPy first to ensure it's available before torch
+RUN pip install numpy==1.24.4
+
+# Install PyTorch CPU version
 RUN pip install torch==2.0.1+cpu torchvision==0.15.2+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
 
-# Copy requirements and install the rest
-COPY requirements.txt ./
+# Copy and install requirements
+COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy application code
+# Copy app
 COPY app.py .
 
 EXPOSE 5000
+
 CMD ["python", "app.py"]