fengmiguoji commited on
Commit
2f6cdee
·
verified ·
1 Parent(s): 15d3a1c

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -3
Dockerfile CHANGED
@@ -22,15 +22,16 @@ WORKDIR /app
22
 
23
  # 设置 bizyair 的缓存目录
24
  ENV BIZYAIR_CACHE_DIR="/tmp/.bizyair_cache"
 
25
  # 设置 TRANSFORMERS 的缓存目录
26
  ENV TRANSFORMERS_CACHE="/tmp/.huggingface_cache"
27
 
28
  # 定义端口
29
  EXPOSE 8188
30
 
31
- # 下载模型等数据到容器内部 (这里只是示例)
32
- # RUN cd /app/models && wget <model_url>
33
- # RUN cd /app/input && wget <input_url>
34
 
35
  # 修改启动命令为使用 CPU
36
  CMD ["python", "main.py", "--listen", "--port", "8188", "--cpu"]
 
22
 
23
  # 设置 bizyair 的缓存目录
24
  ENV BIZYAIR_CACHE_DIR="/tmp/.bizyair_cache"
25
+
26
  # 设置 TRANSFORMERS 的缓存目录
27
  ENV TRANSFORMERS_CACHE="/tmp/.huggingface_cache"
28
 
29
  # 定义端口
30
  EXPOSE 8188
31
 
32
+ # 预先创建缓存目录并赋予权限
33
+ RUN mkdir -p /tmp/.bizyair_cache && chmod -R 777 /tmp/.bizyair_cache
34
+
35
 
36
  # 修改启动命令为使用 CPU
37
  CMD ["python", "main.py", "--listen", "--port", "8188", "--cpu"]