Upload Dockerfile

Dockerfile

@@ -1,47 +1,36 @@
 FROM python:3.10-slim
 
 WORKDIR /app
+COPY . .
 
-# 安装系统依赖
-RUN apt-get update && apt-get install -y \
-    build-essential \
-    libgl1-mesa-glx \
-    libglib2.0-0 \
-    wget \
-    git \
-    && rm -rf /var/lib/apt/lists/*
+# 安装系统依赖（OpenCV 等需要）
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    software-properties-common \
+    libsm6 libxext6 ffmpeg libfontconfig1 libxrender1 libgl1-mesa-glx \
+    curl gcc build-essential
 
-# 创建模型和缓存目录并设置权限
+# 创建模型和缓存目录
 RUN mkdir -p /app/models && \
     chmod 777 /app/models && \
-    mkdir -p /app/.cache/huggingface/hub && \
-    chmod 777 /app/.cache/huggingface/hub && \
     mkdir -p /tmp/iopaint/models && \
     chmod 777 /tmp/iopaint/models
 
-# 复制项目文件
-COPY . /app/
+# 安装 PyTorch CPU 版和依赖
+RUN pip install --no-cache-dir torch==2.0.1 torchvision==0.15.2 -f https://download.pytorch.org/whl/cpu/torch_stable.html
+RUN pip install -r requirements.txt
+RUN pip install -e .
 
-# 确保入口脚本有执行权限
-RUN chmod +x /app/entrypoint.sh
-
-# 安装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
-ENV TRANSFORMERS_CACHE=/app/.cache/huggingface/hub
-ENV HF_HOME=/app/.cache/huggingface
 ENV IOPAINT_MODEL_DIR=/app/models
+ENV HF_ENDPOINT=https://hf-mirror.com
 
-# 暴露端口
 EXPOSE 7860
 
-# 使用入口脚本启动
-CMD ["/app/entrypoint.sh"] 
+# 使用我们的API服务脚本
+CMD ["python", "-m", "iopaint.api_only"]