Upload Dockerfile

Dockerfile

@@ -11,6 +11,10 @@ COPY ./requirements.txt /code/requirements.txt
 # Install the Python packages specified in requirements.txt
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
+# Set the environment variable for the transformers cache
+ENV TRANSFORMERS_CACHE=/code/cache/huggingface/transformers
+ENV HF_HOME=/code/cache/huggingface
+
 # Download the llama-2-7b-chat.ggmlv3.q8_0.bin model file into the container
 # Replace the URL below with the actual URL from where the model file should be downloaded
 RUN wget -O /code/llama-2-7b-chat.ggmlv3.q8_0.bin https://huggingface.co/TheBloke/Llama-2-7B-Chat-GGML/blob/main/llama-2-7b-chat.ggmlv3.q8_0.bin