test2023h5 commited on
Commit
096aced
·
verified ·
1 Parent(s): c4b89ec

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -0
Dockerfile CHANGED
@@ -6,6 +6,9 @@ COPY ./requirements.txt /code/requirements.txt
6
 
7
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
8
 
 
 
 
9
  COPY . .
10
 
11
  CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
 
6
 
7
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
8
 
9
+ # 设置 TRANSFORMERS_CACHE 环境变量
10
+ ENV TRANSFORMERS_CACHE="/code/.cache/huggingface"
11
+
12
  COPY . .
13
 
14
  CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]