Forrest99 commited on
Commit
7b51116
·
verified ·
1 Parent(s): 8e74b06

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +26 -5
Dockerfile CHANGED
@@ -1,12 +1,33 @@
1
  FROM python:3.9-slim
2
 
 
 
 
 
 
 
 
 
 
 
 
3
  WORKDIR /app
4
- COPY . .
5
 
6
- # 先安装依赖确保版本正确
7
- RUN pip install --no-cache-dir -r requirements.txt \
8
- && pip install "sentence-transformers" \
9
- && pip install "torch" --extra-index-url https://download.pytorch.org/whl/cpu
 
 
 
 
 
 
 
 
 
 
 
10
 
11
  EXPOSE 5000
12
  CMD ["python", "app.py"]
 
1
  FROM python:3.9-slim
2
 
3
+ # 创建专用用户和缓存目录
4
+ RUN useradd -m appuser && \
5
+ mkdir -p /app/model-cache && \
6
+ chown -R appuser:appuser /app
7
+
8
+ # 设置环境变量
9
+ ENV TRANSFORMERS_CACHE=/app/model-cache \
10
+ HF_HOME=/app/model-cache
11
+
12
+ # 切换到非root用户
13
+ USER appuser
14
  WORKDIR /app
 
15
 
16
+ # 先安装系统依赖
17
+ RUN apt-get update && apt-get install -y --no-install-recommends \
18
+ gcc \
19
+ python3-dev \
20
+ && rm -rf /var/lib/apt/lists/*
21
+
22
+ # 安装Python依赖
23
+ COPY requirements.txt .
24
+ RUN pip install --user --no-cache-dir -r requirements.txt
25
+
26
+ # 复制应用代码
27
+ COPY app.py .
28
+
29
+ # 设置PATH
30
+ ENV PATH="/home/appuser/.local/bin:${PATH}"
31
 
32
  EXPOSE 5000
33
  CMD ["python", "app.py"]