# Must use a Cuda version 11+
# FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime
FROM pytorch/pytorch:1.13.0-cuda11.6-cudnn8-runtime

WORKDIR /
COPY ./parser /parser
COPY ./configs /configs
RUN mkdir /checkpoints
# Install git
RUN apt-get update && apt-get install -y --no-install-recommends \
    git \
    build-essential

# Install python packages
RUN pip3 install --upgrade pip
ADD requirements.txt requirements.txt
RUN pip3 install -r requirements.txt

# Install cv2 dependencies
RUN apt-get update
RUN apt-get install ffmpeg libsm6 libxext6  -y

# We add the banana boilerplate here
ADD server.py .

# Add your model weight files 
# (in this case we have a python script)
ADD download.py .

# Add your custom app code, init() and inference()
ADD app.py .

ENV PYTHONPATH "${PYTHONPATH}:/parser:/upscaler"

#Alternative to using build args, you can put your token in the next line
#ENV HF_AUTH_TOKEN={token}
RUN python3 download.py

EXPOSE 8000

CMD python3 -u server.py