FROM monius/docker-ai-infra

USER infra
COPY --chown=infra . /opt/run/
RUN echo "av" > /opt/run/ai-infra.txt
RUN cat /opt/run/ai-infra.txt
RUN /opt/dev/bin/pip install datasets
RUN /opt/dev/bin/pip install --no-cache-dir huggingface_hub[hf_transfer] pyarrow
CMD ["ai-infra"]