Dockerfile

# 使用Python的官方镜像
FROM python:3.9-slim

# 设置工作目录
WORKDIR /app

# 复制本地代码到容器中
COPY . /app

# 创建一个非 root 用户
RUN useradd -m myuser
RUN chown -R myuser:myuser /app

# 切换到非 root 用户
USER myuser

# 将 /home/myuser/.local/bin 添加到 PATH
ENV PATH="/home/myuser/.local/bin:${PATH}"

# 创建缓存目录并设置权限
RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache

# 使用 HF_HOME 替代 TRANSFORMERS_CACHE
ENV HF_HOME=/app/.cache

# 安装依赖
RUN pip install --no-cache-dir -r requirements.txt

# 设置容器启动时的命令
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]