i-darrshan commited on
Commit
5ec0b26
1 Parent(s): 7caa513

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +13 -2
Dockerfile CHANGED
@@ -1,2 +1,13 @@
1
- CMD ["docker", "run", "-p", "8001:80", "-d", "--rm", "--name", "huggingface-embedding-server", "ghcr.io/huggingface/text-embeddings-inference:cpu-0.3.0", "--model-id", "BAAI/bge-small-en-v1.5", "--revision", "-main"]
2
- EXPOSE 80
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Use the base image for Hugging Face text embeddings inference
2
+ FROM ghcr.io/huggingface/text-embeddings-inference:cpu-0.3.0
3
+
4
+ # Define the entrypoint for the container
5
+ ENTRYPOINT ["text-embeddings-inference"]
6
+
7
+ # Set default arguments for the entrypoint
8
+ CMD ["--model-id", "BAAI/bge-small-en-v1.5", "--revision", "-main", "--port", "7860"]
9
+
10
+ # Expose the port on which the service will run (Hugging Face Spaces uses port 7860)
11
+ EXPOSE 7860
12
+
13
+ CMD ["docker", "run", "-p", "8001:7860", "-d", "--rm", "--name", "huggingface-embedding-server", "huggingface-embedding-server"]