QUIET added

3x3cut0r · Feb 18, 2024 · c76d376 · c76d376
1 parent a87f02e
commit c76d376
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/privategpt/Dockerfile b/privategpt/Dockerfile
@@ -122,7 +122,8 @@ ENV ENV_NAME="prod" \
     CORS_ALLOW_HEADERS="*" \
     AUTH_ENABLED="false" \
     AUTH_USERNAME="secret" \
-    AUTH_SECRET="Basic c2VjcmV0OmtleQ=="
+    AUTH_SECRET="Basic c2VjcmV0OmtleQ==" \
+    QUIET="false"
 
 VOLUME /home/worker/app/local_data
 VOLUME /home/worker/app/models

diff --git a/privategpt/docker-entrypoint.sh b/privategpt/docker-entrypoint.sh
@@ -134,7 +134,12 @@ sed -i "75s|^.*$|  model: ${OLLAMA_MODEL:-"mistral:latest"}|" /home/worker/app/s
 ############################
 
 # run privategpt setup to download models
-/home/worker/app/.venv/bin/python scripts/setup
+if [ "$QUIET" = "true" ]; then
+  /home/worker/app/.venv/bin/python scripts/setup > /dev/null 2>&1
+else
+  /home/worker/app/.venv/bin/python scripts/setup
+fi
+
 
 # if started without args, run privategpt
 if [ "$#" = "0" ]; then