cigol123 commited on
Commit
9a04092
·
verified ·
1 Parent(s): 58aeb8f

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -2,11 +2,11 @@
2
  FROM python:3.9-slim
3
 
4
  # Set environment variables for Hugging Face cache
5
- ENV TRANSFORMERS_CACHE=/app/.cache
6
- ENV HF_DATASETS_CACHE=/app/.cache
7
 
8
  # Create the cache directory and set permissions
9
- RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
10
 
11
  # Set the working directory in the container
12
  WORKDIR /app
 
2
  FROM python:3.9-slim
3
 
4
  # Set environment variables for Hugging Face cache
5
+ ENV TRANSFORMERS_CACHE=/tmp/.cache
6
+ ENV HF_DATASETS_CACHE=/tmp/.cache
7
 
8
  # Create the cache directory and set permissions
9
+ RUN mkdir -p /tmp/.cache && chmod -R 777 /tmp/.cache
10
 
11
  # Set the working directory in the container
12
  WORKDIR /app