fiewolf1000 commited on
Commit
849bc37
·
verified ·
1 Parent(s): ce23ec0

Create Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +28 -0
Dockerfile ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # 1. 基础镜像:选择 Python 3.10(适配 transformers、torch 等库的稳定版本)
2
+ FROM python:3.10-slim
3
+
4
+ # 2. 设置工作目录(避免文件混乱)
5
+ WORKDIR /app
6
+
7
+ # 3. 安装系统依赖(解决 torch、transformers 依赖的底层库问题)
8
+ RUN apt-get update && apt-get install -y --no-install-recommends \
9
+ gcc \
10
+ g++ \
11
+ && rm -rf /var/lib/apt/lists/*
12
+
13
+ # 4. 复制依赖文件并安装 Python 库(先复制 requirements.txt 可利用 Docker 缓存,加速后续构建)
14
+ COPY requirements.txt .
15
+ RUN pip install --no-cache-dir -r requirements.txt
16
+
17
+ # 5. 复制所有应用代码(包括 app.py)
18
+ COPY . .
19
+
20
+ # 6. 设置环境变量(可选,用于指定模型加载设备、日志级别等)
21
+ ENV PYTHONUNBUFFERED=1 \
22
+ TORCH_DEVICE=cuda # 若 Spaces 分配 GPU 则自动使用,无 GPU 会 fallback 到 CPU
23
+
24
+ # 7. 暴露端口(Hugging Face Spaces 会自动映射该端口,需与 app.py 中 uvicorn 端口一致)
25
+ EXPOSE 7860
26
+
27
+ # 8. 启动命令(启动 uvicorn 服务,确保进程前台运行)
28
+ CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]