Update Dockerfile

Dockerfile

@@ -13,12 +13,15 @@ RUN chown -R myuser:myuser /app /home/myuser
 # Switch to the non-root user
 USER myuser
 
+# Set the Hugging Face token as a build argument
+ARG HUGGINGFACE_TOKEN
+
+# Run huggingface-cli login with the provided token
+RUN huggingface-cli login --token $HUGGINGFACE_TOKEN
+
 # Copy the entire contents of the local directory into the container
 COPY . .
 
-# Download the model file
-RUN wget https://huggingface.co/TheBloke/Llama-2-7B-Chat-GGML/resolve/main/llama-2-7b-chat.ggmlv3.q8_0.bin
-
 # Install chainlit and add it to PATH
 RUN pip install chainlit --user
 
@@ -31,5 +34,5 @@ RUN pip install -r requirements.txt
 # Expose port 7860 internally in the container
 EXPOSE 7860
 
-# Run the ChainlIt command
+# Run the ChainLit command
 CMD ["chainlit", "run", "model.py", "-w", "--port", "7860"]