Forrest99 commited on
Commit
7efcb5c
·
verified ·
1 Parent(s): b6af10b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -18
Dockerfile CHANGED
@@ -1,26 +1,12 @@
1
- # 使用官方 Python 镜像
2
  FROM python:3.9-slim
3
 
4
- # 设置缓存目录(必须用 root 用户创建)
5
- RUN mkdir -p /app/.cache/huggingface && \
6
- chmod 777 /app /app/.cache /app/.cache/huggingface
7
-
8
- # 创建非 root 用户并切换
9
- RUN useradd -m -u 1000 appuser
10
- USER appuser
11
-
12
- # 设置工作目录和环境变量
13
- WORKDIR /app
14
- ENV HF_HOME=/app/.cache/huggingface \
15
- TRANSFORMERS_CACHE=/app/.cache/huggingface \
16
- HUGGINGFACE_HUB_CACHE=/app/.cache/huggingface
17
 
18
  # 安装依赖
19
- COPY requirements.txt .
20
- RUN pip install --no-cache-dir -r requirements.txt
21
 
22
  # 复制代码
23
- COPY . .
24
 
25
- # 启动命令
26
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
 
1
  FROM python:3.9-slim
2
 
3
+ # 创建缓存目录并赋予权限
4
+ RUN mkdir -p /app/.cache/huggingface && chmod 777 -R /app
 
 
 
 
 
 
 
 
 
 
 
5
 
6
  # 安装依赖
7
+ RUN pip install "fastapi>=0.68" "uvicorn>=0.15" "torch>=1.9" "transformers>=4.11"
 
8
 
9
  # 复制代码
10
+ COPY app.py .
11
 
 
12
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]