FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
ENV DEBIAN_FRONTEND=noninteractive

RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user \
    PATH=/home/user/.local/bin:${PATH}
WORKDIR ${HOME}/app

COPY --chown=1000 . ${HOME}/app
RUN pip install -r ${HOME}/app/requirements.txt && \
    pip install fairseq2 --pre --extra-index-url https://fair.pkg.atmeta.com/fairseq2/pt2.1.0/cu121 && \
    pip install ${HOME}/app/whl/seamless_communication-1.0.0-py3-none-any.whl
# This will cache the model into the docker image
RUN python -u translator.py
CMD ["python", "main.py"]