From 54c29f8bd3c76286708d29a6cda494f9063e2b8b Mon Sep 17 00:00:00 2001
From: Jun Siang Cheah <git@jscheah.me>
Date: Tue, 21 May 2024 21:28:38 +0100
Subject: [PATCH] fix: chown /app/backend/data after preloading models

---
 Dockerfile       | 3 ++-
 backend/start.sh | 2 --
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index dee049fb4..c2c42aa17 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -132,7 +132,8 @@ RUN pip3 install uv && \
     uv pip install --system -r requirements.txt --no-cache-dir && \
     python -c "import os; from sentence_transformers import SentenceTransformer; SentenceTransformer(os.environ['RAG_EMBEDDING_MODEL'], device='cpu')" && \
     python -c "import os; from faster_whisper import WhisperModel; WhisperModel(os.environ['WHISPER_MODEL'], device='cpu', compute_type='int8', download_root=os.environ['WHISPER_MODEL_DIR'])"; \
-    fi
+    fi; \
+    chown -R $UID:$GID /app/backend/data/
 
 
 
diff --git a/backend/start.sh b/backend/start.sh
index 209f3e31b..a36feadd9 100755
--- a/backend/start.sh
+++ b/backend/start.sh
@@ -31,7 +31,6 @@ if [ "$USE_CUDA_DOCKER" = "true" ]; then
 fi
 
 
-# HFSPACE:START
 # Check if SPACE_ID is set, if so, configure for space
 if [ -n "$SPACE_ID" ]; then
   echo "Configuring for HuggingFace Space deployment"
@@ -56,6 +55,5 @@ if [ -n "$SPACE_ID" ]; then
   kill $webui_pid
   export WEBUI_URL=${SPACE_HOST}
 fi
-# HFSPACE:END
 
 WEBUI_SECRET_KEY="$WEBUI_SECRET_KEY" exec uvicorn main:app --host "$HOST" --port "$PORT" --forwarded-allow-ips '*'