SAN-Demo / Dockerfile
Mendel192's picture
Update Dockerfile
f480abf
FROM mendelxu/pytorch:d2_nvcr_2008
# Set up a new user named "user" with user ID 1000
RUN useradd -m -u 1000 user
# Switch to the "user" user
USER user
# Set home to the user's home directory
ENV HOME=/home/user \
PATH=/home/user/.local/bin:$PATH
# Set the working directory to the user's home directory
WORKDIR $HOME/app
RUN --mount=type=secret,id=HF_TOKEN,mode=0444,required=true
# clone from the newest code.
RUN ls -l $HOME/app
RUN git init && git remote add origin https://github.com/MendelXu/SAN.git
RUN git pull origin main
# gradio
RUN pip install gradio
ENV GRADIO_SERVER_NAME=0.0.0.0
EXPOSE 7860
RUN echo "gradio app.py">>run.sh
CMD ["script","-c","sh run.sh","/dev/null"]