Mridul2003 commited on
Commit
352bac3
·
verified ·
1 Parent(s): 8adfd01

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -5
Dockerfile CHANGED
@@ -1,12 +1,10 @@
1
  FROM python:3.10
2
 
3
  # Create a writable cache directory
4
- RUN mkdir -p /app/cache
5
 
6
- # Set environment variables to redirect cache
7
- ENV TRANSFORMERS_CACHE=/app/cache \
8
- HF_HOME=/app/cache \
9
- HF_DATASETS_CACHE=/app/cache
10
 
11
  WORKDIR /app
12
 
 
1
  FROM python:3.10
2
 
3
  # Create a writable cache directory
4
+ RUN mkdir -p /app/cache && chmod -R 777 /app/cache
5
 
6
+ # Set environment variable (HF_HOME is preferred over TRANSFORMERS_CACHE)
7
+ ENV HF_HOME=/app/cache
 
 
8
 
9
  WORKDIR /app
10