codebertBase / Dockerfile
Forrest99's picture
Update Dockerfile
47336a8 verified
raw
history blame
664 Bytes
# 使用官方 Python 镜像
FROM python:3.9-slim
# 设置缓存目录(必须用 root 用户创建)
RUN mkdir -p /app/.cache/huggingface && \
chmod 777 /app /app/.cache /app/.cache/huggingface
# 创建非 root 用户并切换
RUN useradd -m -u 1000 appuser
USER appuser
# 设置工作目录和环境变量
WORKDIR /app
ENV HF_HOME=/app/.cache/huggingface \
TRANSFORMERS_CACHE=/app/.cache/huggingface \
HUGGINGFACE_HUB_CACHE=/app/.cache/huggingface
# 安装依赖
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
# 复制代码
COPY . .
# 启动命令
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]