FROM python:3.9-slim

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

# 安装依赖
RUN pip install "fastapi>=0.68" "uvicorn>=0.15" "torch>=1.9" "transformers>=4.11"

# 复制代码
COPY app.py .

CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]