prodigy-rlhf-ranking / Dockerfile
wesslen's picture
update Dockerfile for sqlite3, user
9773321
raw
history blame
1.32 kB
FROM python:3.9
#COPY requirements.txt /app/
WORKDIR /app
RUN sudo apt-get install -y sqlite3 libsqlite3-dev
RUN --mount=type=secret,id=LICENSE_KEY,mode=0444,required=true \
pip install --upgrade pip \
&& pip install prodigy -f https://$(cat /run/secrets/LICENSE_KEY)@download.prodi.gy
# && pip install --trusted-host pypi.python.org -r requirements.txt
# COPY wheel/prodigy-xxx-linux_x86_64.whl ./wheel/
# RUN pip install wheel/prodigy-xxx-linux_x86_64.whl \
# && rm -rf wheel/prodigy-xxx-linux_x86_64.whl
RUN python -m spacy download en_core_web_sm
COPY prodigy.json .
COPY data ./data/
# 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
# Copy the current directory contents into the container at $HOME/app setting the owner to the user
COPY --chown=user . $HOME/app
ENV PRODIGY_HOME /app
ENV PRODIGY_LOGGING "verbose"
ENV PRODIGY_ALLOWED_SESSIONS "user1,user2"
ENV PRODIGY_BASIC_AUTH_USER "admin"
ENV PRODIGY_BASIC_AUTH_PASS "password"
EXPOSE 7860
CMD python -m prodigy ner.manual ner_news en_core_web_sm ./data/dataset.jsonl --label PERSON,ORG,PRODUCT