repainting_3d_assets / Dockerfile
toshas's picture
Initial commit
c184ad2
FROM toshas/repainting_3d_assets:v2
ENTRYPOINT []
ENV DEBIAN_FRONTEND=noninteractive
ENV REPAINTING3D_DO_NOT_MANAGE_HFCACHE=1
ENV TRANSFORMERS_CACHE=/data/hfcache
ENV HF_DATASETS_CACHE=/data/hfcache
ENV HF_HOME=/data/hfcache
RUN mkdir -p ${HF_HOME}
RUN apt-get -y update --fix-missing && \
apt-get install -y \
python3 \
python3-pip \
&& \
apt-get clean
WORKDIR ${ROOT_PROJECT}
RUN git clone https://github.com/kongdai123/repainting_3d_assets.git && \
rm -rf repainting_3d_assets/.git repainting_3d_assets/assets repainting_3d_assets/doc && \
rm -rf code && \
mv repainting_3d_assets code
COPY ./requirements.txt ./
RUN pip3 install --no-cache-dir --upgrade -r requirements.txt
RUN useradd -m -u 1000 user
RUN chown -R user ${HF_HOME}
USER user
ENV HOME=/home/user \
PATH=/home/user/.local/bin:$PATH
ENV PYTHONUNBUFFERED=1
WORKDIR ${HOME}/app
COPY --chown=user . ${HOME}/app
CMD ["python3", "main.py"]