Update Dockerfile

Dockerfile

@@ -7,8 +7,9 @@ ENV HF_HOME=/home/user/.cache/huggingface
 
 WORKDIR /app
 
-# Use the Hugging Face token from the environment variable
+# Pass the Hugging Face token as an environment variable
 ARG HF_TOKEN
+ENV HF_TOKEN=${HF_TOKEN}
 RUN mkdir -p $HF_HOME && \
     echo "$HF_TOKEN" > $HF_HOME/token
 
@@ -16,10 +17,10 @@ RUN mkdir -p $HF_HOME && \
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-# Pre-load the model
+# Pre-load the model using the updated `token` argument
 RUN python -c "from transformers import AutoTokenizer, AutoModelForCausalLM; \
-    AutoTokenizer.from_pretrained('meta-llama/Llama-3.1-8B-Instruct', use_auth_token=True); \
-    AutoModelForCausalLM.from_pretrained('meta-llama/Llama-3.1-8B-Instruct', use_auth_token=True)"
+    AutoTokenizer.from_pretrained('meta-llama/Llama-3.1-8B-Instruct', token='$HF_TOKEN'); \
+    AutoModelForCausalLM.from_pretrained('meta-llama/Llama-3.1-8B-Instruct', token='$HF_TOKEN')"
 
 # Copy app files
 COPY --chown=user . /app