Update Dockerfile

Dockerfile

@@ -5,20 +5,17 @@ RUN apt-get update && apt-get install -y \
     libgl1 \
     libglib2.0-0 \
     wget \
-    git \
     && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
 
-# Install Python dependencies
+# Download pretrained model
+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 requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Download models and setup basicsr
-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 && \
-    python -c "from basicsr.models import create_model; create_model('RealESRGAN_x4plus')"
-
 COPY app.py .
 
 EXPOSE 5000