Sanket17 commited on
Commit
5a7daf7
·
verified ·
1 Parent(s): 2cfebfe

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +7 -0
Dockerfile CHANGED
@@ -4,6 +4,13 @@ WORKDIR /app
4
 
5
  COPY ./ /app
6
 
 
 
 
 
 
 
 
7
  RUN pip install -r requirements.txt
8
 
9
  RUN pip install "fastapi[standard]"
 
4
 
5
  COPY ./ /app
6
 
7
+ ENV HF_HOME=/app/cache
8
+ ENV TRANSFORMERS_CACHE=/app/cache
9
+ ENV TORCH_HOME=/app/cache
10
+
11
+ RUN mkdir -p /app/cache && chmod -R 777 /app/cache
12
+
13
+
14
  RUN pip install -r requirements.txt
15
 
16
  RUN pip install "fastapi[standard]"