cINAWGD commited on
Commit
cca71ce
·
verified ·
1 Parent(s): e8a949b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -26,6 +26,8 @@ WORKDIR /app
26
  COPY . .
27
 
28
  EXPOSE 7860
 
 
29
 
30
  RUN pipenv run pip install fastapi==0.109.2 uvicorn==0.27.0.post1 torch==2.3.0 transformers==4.37.2 pillow==10.2.0 httpx==0.26.0 pydantic==2.6.1
31
 
 
26
  COPY . .
27
 
28
  EXPOSE 7860
29
+ ENV TRANSFORMERS_CACHE="/tmp/huggingface_cache"
30
+ ENV HF_HOME="/tmp/huggingface_cache"
31
 
32
  RUN pipenv run pip install fastapi==0.109.2 uvicorn==0.27.0.post1 torch==2.3.0 transformers==4.37.2 pillow==10.2.0 httpx==0.26.0 pydantic==2.6.1
33