awinml's picture
Update Dockerfile
ca48187 verified
raw
history blame
1.04 kB
# Use the official Python base image
FROM ghcr.io/huggingface/text-embeddings-inference:cpu-0.5
# Install required packages
RUN apt-get update && \
apt-get install -y \
git \
git-lfs
# Enable git-lfs in the cloned repository
RUN git lfs install
# 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
# Download model from Hugging Face
RUN git clone https://huggingface.co/BAAI/bge-small-en-v1.5
# Copy the current directory contents into the container at $HOME/app setting the owner to the user
COPY --chown=user . $HOME/app/bge-small-en-v1.5
# Set the default command to run your app
CMD ["--model-id", "BAAI/bge-small-en-v1.5", "--port", "7860"]