guge123 commited on
Commit
7e1811b
1 Parent(s): 604f5ef

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -6,7 +6,7 @@ RUN useradd --create-home --shell /bin/bash myuser
6
  WORKDIR /home/myuser/app
7
 
8
  # 安装所需的库
9
- RUN pip install --no-cache-dir torch==1.9.0 transformers flask
10
 
11
  # 设置TRANSFORMERS_CACHE环境变量
12
  ENV TRANSFORMERS_CACHE /home/myuser/.cache/huggingface
@@ -22,3 +22,4 @@ CMD ["python", "app.py"]
22
 
23
 
24
 
 
 
6
  WORKDIR /home/myuser/app
7
 
8
  # 安装所需的库
9
+ RUN pip install --no-cache-dir torch==1.9.0+cpu transformers flask
10
 
11
  # 设置TRANSFORMERS_CACHE环境变量
12
  ENV TRANSFORMERS_CACHE /home/myuser/.cache/huggingface
 
22
 
23
 
24
 
25
+