Update Dockerfile

Dockerfile

@@ -10,13 +10,15 @@ RUN apt-get update && apt-get install -y \
 
 WORKDIR /app
 
-# 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
-
+# Install Python dependencies
 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