Dockerfile

FROM python:3.10-slim

WORKDIR /app

# 安装系统依赖
RUN apt-get update && apt-get install -y \
    build-essential \
    libgl1-mesa-glx \
    libglib2.0-0 \
    wget \
    git \
    && rm -rf /var/lib/apt/lists/*

# 复制项目文件
COPY . /app/

# 安装Python依赖和项目
RUN pip install --no-cache-dir -r requirements.txt
RUN pip install --no-cache-dir -e .

# 环境变量设置
ENV IOPAINT_HOST=0.0.0.0
ENV IOPAINT_PORT=7860
ENV IOPAINT_MODEL=lama
ENV IOPAINT_LOW_MEM=true
ENV IOPAINT_DEVICE=cpu
ENV PYTHONUNBUFFERED=1

# 创建模型目录
RUN mkdir -p /root/.cache/iopaint/models

# 暴露端口
EXPOSE 7860

# 启动后端API服务
CMD ["python", "-m", "iopaint.api_only"]