Update Dockerfile

Dockerfile

@@ -6,6 +6,7 @@ ENV PYTHONUNBUFFERED=1
 ENV HOME=/home/appuser
 ENV APP_HOME=/home/appuser/app
 ENV PATH="/home/appuser/.local/bin:${PATH}"
+ENV HF_HOME="/home/appuser/.cache/huggingface"
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
@@ -28,17 +29,20 @@ WORKDIR $APP_HOME
 # Copy the application files
 COPY --chown=appuser:appuser . $APP_HOME/
 
-# Download the model
-RUN mkdir -p $APP_HOME/models && \
-    wget -O $APP_HOME/models/llama-model.gguf https://huggingface.co/bartowski/Meta-Llama-3.1-8B-Instruct-GGUF/resolve/main/Meta-Llama-3.1-8B-Instruct-IQ2_M.gguf
+# Create necessary directories
+RUN mkdir -p $APP_HOME/.chainlit/files && \
+    mkdir -p $APP_HOME/mydb && \
+    mkdir -p $APP_HOME/models && \
+    mkdir -p $HF_HOME
+
+# Download the model using huggingface-cli
+RUN pip install --user --no-cache-dir huggingface_hub && \
+    huggingface-cli download bartowski/Meta-Llama-3.1-8B-Instruct-GGUF Meta-Llama-3.1-8B-Instruct-IQ2_M.gguf --local-dir $APP_HOME/models/ && \
+    mv $APP_HOME/models/Meta-Llama-3.1-8B-Instruct-IQ2_M.gguf $APP_HOME/models/llama-model.gguf
 
 # Install Python dependencies
 RUN pip install --user --no-cache-dir -r requirements.txt
 
-# Create necessary directories with correct permissions
-RUN mkdir -p $APP_HOME/.chainlit/files && \
-    mkdir -p $APP_HOME/mydb
-
 # Create a chainlit.md file
 RUN echo "# Welcome to RAG Chainlit Application! 👋\n\nThis is a Retrieval-Augmented Generation application using Llama.cpp." > $APP_HOME/chainlit.md