prodigy-rlhf-ranking / Dockerfile
wesslen's picture
add in user auth; venv
989aa4d
raw
history blame
1.55 kB
FROM python:3.9
#COPY requirements.txt /app/
WORKDIR /app
# # 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
RUN --mount=type=secret,id=LICENSE_KEY,mode=0444,required=true \
python -m venv venv \
&& source venv/bin/activate \
&& 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
RUN chmod 777 .
COPY prodigy.json .
COPY data ./data/
ENV PRODIGY_HOME /app
ENV PRODIGY_LOGGING "verbose"
ENV PRODIGY_ALLOWED_SESSIONS "user1,user2"
RUN --mount=type=secret,id=AUTH_USER,mode=0444,required=true \
ENV PRODIGY_BASIC_AUTH_USER $(cat /run/secrets/AUTH_USER)
RUN --mount=type=secret,id=AUTH_PASS,mode=0444,required=true \
ENV PRODIGY_BASIC_AUTH_PASS $(cat /run/secrets/AUTH_PASS)
EXPOSE 7860
CMD python -m prodigy ner.manual ner_news en_core_web_sm ./data/dataset.jsonl --label PERSON,ORG,PRODUCT