FROM python:3.9-slim-bullseye

RUN apt-get update
RUN apt-get install git ffmpeg libsm6 libxext6 wget -y

ENV LMDB_FORCE_CFFI=1
#RUN pip install cython

COPY requirements.txt .
RUN pip install --upgrade pip
RUN pip install -r requirements.txt

# Create working directory
RUN mkdir -p /usr/src/app
WORKDIR /usr/src/app

# Copy model
COPY weights /usr/src/app/weights
#COPY gfpgan /usr/src/app/gfpgan

# Copy source code
COPY inference.py /usr/src/app

ENTRYPOINT ["python3", "/usr/src/app/inference.py", "-n", "RealESRGAN_x4plus_anime_6B", "-i", "/in", "-o", "/out", "-s", "4", "--fp32"]