taowang1993 commited on
Commit
7f0dd55
·
verified ·
1 Parent(s): d90d6e0

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +34 -8
Dockerfile CHANGED
@@ -1,14 +1,40 @@
1
- # Use a prebuilt image instead of building from scratch
2
  FROM ghcr.io/open-webui/open-webui:main
3
 
4
- # Set the data directory environment variable
5
- ENV DATA_DIR=/app/backend/data
6
 
7
- # Ensure the data directory exists and has the correct ownership and permissions
8
- RUN mkdir -p ${DATA_DIR} && chown -R 1000:1000 ${DATA_DIR}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
 
10
- # Expose the web server port
 
 
 
 
 
11
  EXPOSE 8080
12
 
13
- # Start the application
14
- CMD ["bash", "start.sh"]
 
 
 
 
1
+ # Use the prebuilt image
2
  FROM ghcr.io/open-webui/open-webui:main
3
 
4
+ # Set the port to 8080
5
+ ENV PORT=8080
6
 
7
+ # Set all necessary environment variables
8
+ ENV ENV=prod \
9
+ USE_OLLAMA_DOCKER=${USE_OLLAMA} \
10
+ USE_CUDA_DOCKER=${USE_CUDA} \
11
+ USE_CUDA_DOCKER_VER=${USE_CUDA_VER} \
12
+ USE_EMBEDDING_MODEL_DOCKER=${USE_EMBEDDING_MODEL} \
13
+ USE_RERANKING_MODEL_DOCKER=${USE_RERANKING_MODEL} \
14
+ OLLAMA_BASE_URL="/ollama" \
15
+ OPENAI_API_BASE_URL="" \
16
+ OPENAI_API_KEY="" \
17
+ WEBUI_SECRET_KEY="" \
18
+ SCARF_NO_ANALYTICS=true \
19
+ DO_NOT_TRACK=true \
20
+ ANONYMIZED_TELEMETRY=false \
21
+ WHISPER_MODEL="base" \
22
+ WHISPER_MODEL_DIR="/app/backend/data/cache/whisper/models" \
23
+ RAG_EMBEDDING_MODEL="$USE_EMBEDDING_MODEL_DOCKER" \
24
+ RAG_RERANKING_MODEL="$USE_RERANKING_MODEL_DOCKER" \
25
+ SENTENCE_TRANSFORMERS_HOME="/app/backend/data/cache/embedding/models" \
26
+ HF_HOME="/app/backend/data/cache/embedding/models"
27
 
28
+ # Create data directories and set permissions
29
+ RUN mkdir -p /app/backend/data/cache/whisper/models && \
30
+ mkdir -p /app/backend/data/cache/embedding/models && \
31
+ chown -R 1000:1000 /app/backend/data
32
+
33
+ # Expose the port
34
  EXPOSE 8080
35
 
36
+ # Use the specified user and group
37
+ USER 1000:1000
38
+
39
+ # Command to start the application
40
+ CMD ["bash", "start.sh"]