Spaces:
Sleeping
Sleeping
Create Dockerfile
Browse files- 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"]
|