Create Dockerfile

Dockerfile

@@ -0,0 +1,49 @@
+FROM python:3.10-slim
+
+# System dependencies (no CUDA requirements)
+RUN apt-get update && apt-get install -y \
+    cmake \
+    build-essential \
+    libopenblas-dev \
+    libx11-dev \
+    libgl1-mesa-glx \
+    git \
+    wget \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install CPU-specific PyTorch first
+RUN pip install --upgrade pip && \
+    pip install torch==2.0.1 torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cpu
+
+# Copy requirements
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Install dlib with CUDA explicitly disabled
+RUN CMAKE_ARGS="-DDLIB_USE_CUDA=0" pip install dlib==19.24.1
+
+# Clone CodeFormer
+RUN git clone https://github.com/sczhou/CodeFormer
+WORKDIR /CodeFormer
+
+# Create required directory structure
+RUN mkdir -p basicsr/basicsr && \
+    echo "# BasicSR" > basicsr/README.md && \
+    echo "1.3.2" > basicsr/basicsr/VERSION
+
+# Install BasicSR with CPU support
+RUN cd basicsr && \
+    python setup.py develop && \
+    cd ..
+
+# Download models
+RUN python scripts/download_pretrained_models.py facelib
+RUN python scripts/download_pretrained_models.py CodeFormer
+
+# Set environment variables
+ENV PYTHONPATH=/CodeFormer:$PYTHONPATH
+
+WORKDIR /
+COPY app.py .
+
+CMD ["python", "app.py"]