Update Dockerfile

Dockerfile

@@ -1,28 +1,23 @@
-FROM python:3.10
-
-WORKDIR /app
+FROM python:3.9-slim
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
-    libgl1 libglib2.0-0 libsm6 libxext6 libxrender-dev && \
-    rm -rf /var/lib/apt/lists/*
-
-# Upgrade pip
-RUN pip install --upgrade pip
+    libgl1 \
+    libglib2.0-0 \
+    wget \
+    git \
+    && rm -rf /var/lib/apt/lists/*
 
-# Install NumPy first to ensure it's available before torch
-RUN pip install numpy==1.24.4
+WORKDIR /app
 
-# 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
+# Download pretrained models
+RUN mkdir -p /app/weights && \
+    wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.pth -P /app/weights
 
-# Copy and install requirements
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy app
 COPY app.py .
 
 EXPOSE 5000
-
-CMD ["python", "app.py"]
+CMD ["gunicorn", "--bind", "0.0.0.0:5000", "--timeout", "600", "app:app"]