huanbao commited on
Commit
d3fdbea
·
verified ·
1 Parent(s): 7cdf179

Create Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +27 -0
Dockerfile ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.10-slim
2
+
3
+ WORKDIR /app
4
+
5
+ # 1. 安装 git 和 git-lfs (下载模型需要)
6
+ RUN apt-get update && apt-get install -y git git-lfs
7
+
8
+ # 2. 复制依赖并安装
9
+ COPY requirements.txt .
10
+ RUN pip install --no-cache-dir -r requirements.txt
11
+
12
+ # 3. 核心步骤:直接在服务器端下载模型到 /app/models 目录
13
+ # 使用 HF 镜像站加速(虽然在HF内部不需要,但加上保险)
14
+ # 只下载 safetensors 等核心文件,排除大文件避免镜像过大(如果需要)
15
+ RUN pip install huggingface_hub
16
+ RUN huggingface-cli download --resume-download BAAI/bge-m3 --local-dir /app/models/bge-m3
17
+ RUN huggingface-cli download --resume-download BAAI/bge-reranker-v2-m3 --local-dir /app/models/bge-reranker
18
+
19
+ # 4. 复制其余代码
20
+ COPY . .
21
+
22
+ # 5. 设置权限和环境变量
23
+ RUN mkdir -p /app/cache && chmod 777 /app/cache
24
+ ENV HF_HOME=/app/cache
25
+
26
+ EXPOSE 7860
27
+ CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]