awinml commited on
Commit
6195fd3
·
verified ·
1 Parent(s): fed254c

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -10
Dockerfile CHANGED
@@ -1,6 +1,5 @@
1
- # Use the official Python base image
2
- FROM ghcr.io/huggingface/text-embeddings-inference:cpu-0.5
3
-
4
 
5
  # Install required packages
6
  RUN apt-get update && \
@@ -24,15 +23,11 @@ ENV HOME=/home/user \
24
  # Set the working directory to the user's home directory
25
  WORKDIR $HOME/app
26
 
27
- # Copy the current directory contents into the container at $HOME/app setting the owner to the user
28
- COPY --chown=user . $HOME/app
29
-
30
  # Download model from Hugging Face
31
- RUN git clone https://huggingface.co/BAAI/bge-small-en-v1.5
32
 
33
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
34
- COPY --chown=user . $HOME/app/bge-small-en-v1.5
35
 
36
  # Set the default command to run your app
37
- CMD ["--model-id", "bge-small-en-v1.5", "--port", "7860"]
38
-
 
1
+ # Use the official TEI Image
2
+ FROM ghcr.io/huggingface/text-embeddings-inference:cpu-1.1
 
3
 
4
  # Install required packages
5
  RUN apt-get update && \
 
23
  # Set the working directory to the user's home directory
24
  WORKDIR $HOME/app
25
 
 
 
 
26
  # Download model from Hugging Face
27
+ RUN git clone https://huggingface.co/BAAI/bge-large-en-v1.5
28
 
29
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
30
+ COPY --chown=user . $HOME/app
31
 
32
  # Set the default command to run your app
33
+ CMD ["--model-id", "bge-large-en-v1.5", "--port", "7860"]