darshan8950 commited on
Commit
1f33461
·
verified ·
1 Parent(s): 88e7405

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -6,8 +6,9 @@ COPY ./requirements.txt /code/requirements.txt
6
 
7
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
8
 
9
- # Create cache directory with proper permissions
10
- RUN mkdir -p /code/.cache/huggingface && chmod -R 777 /code/.cache
 
11
 
12
  COPY . .
13
 
 
6
 
7
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
8
 
9
+ # Create a writable cache directory
10
+ RUN mkdir /.cache && chmod -R 777 /.cache
11
+ ENV TRANSFORMERS_CACHE=/.cache
12
 
13
  COPY . .
14