Dockerfile

# 使用官方的 Python Docker 镜像作为基础镜像
FROM python:3.8-slim

# 创建一个非特权用户并设置工作目录
RUN useradd --create-home --shell /bin/bash myuser
WORKDIR /home/myuser/app

# 安装所需的库
RUN pip install --no-cache-dir torch==1.9.0 transformers flask

# 设置TRANSFORMERS_CACHE环境变量
ENV TRANSFORMERS_CACHE /home/myuser/.cache/huggingface

# 复制当前目录内容到容器的 /app 目录下
COPY . .

# 更改文件夹所有者为非特权用户
RUN chown -R myuser:myuser /home/myuser

# 指定默认命令，运行 Python 脚本
CMD ["python", "app.py"]