Marroco93 commited on
Commit
b86a744
1 Parent(s): 1f94144

no message

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -4
Dockerfile CHANGED
@@ -5,12 +5,11 @@ WORKDIR /code
5
  COPY ./requirements.txt /code/requirements.txt
6
 
7
  # Set the cache directory for transformers and Hugging Face Hub within the container
8
- ENV TRANSFORMERS_CACHE=/code/cache/huggingface/transformers
9
  ENV HF_HOME=/code/cache/huggingface/hub
10
 
11
- # Create the cache directories
12
- RUN mkdir -p $TRANSFORMERS_CACHE
13
- RUN mkdir -p $HF_HOME
14
 
15
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
16
 
 
5
  COPY ./requirements.txt /code/requirements.txt
6
 
7
  # Set the cache directory for transformers and Hugging Face Hub within the container
 
8
  ENV HF_HOME=/code/cache/huggingface/hub
9
 
10
+ # Create the cache directories and set permissions
11
+ RUN mkdir -p $HF_HOME && \
12
+ chown -R 1000:1000 $HF_HOME
13
 
14
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
15