Upload Dockerfile

Dockerfile

@@ -1,39 +1,43 @@
-FROM python:3.10-slim
+FROM python:3.10
 
 WORKDIR /app
-COPY . .
 
-# 安装系统依赖（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/checkpoints && \
-    chmod 777 /app/models/checkpoints && \
-    mkdir -p /tmp/iopaint/models && \
-    chmod 777 /tmp/iopaint/models && \
-    mkdir -p /tmp/iopaint/checkpoints && \
-    chmod 777 /tmp/iopaint/checkpoints
-
-# 安装 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 apt-get update && apt-get install -y \
+    build-essential \
+    curl \
+    git \
+    ffmpeg \
+    libsm6 \
+    libxext6 \
+    && rm -rf /var/lib/apt/lists/*
+
+# 创建并设置模型和缓存目录
+RUN mkdir -p /app/models /tmp/iopaint/cache \
+    && chmod -R 777 /app/models /tmp/iopaint
 
 # 设置环境变量
-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 IOPAINT_MODEL_DIR=/app/models
-ENV HF_ENDPOINT=https://hf-mirror.com
-ENV TORCH_HOME=/app/models
+ENV PYTHONUNBUFFERED=1 \
+    PYTHONDONTWRITEBYTECODE=1 \
+    PIP_NO_CACHE_DIR=1 \
+    IOPAINT_MODEL=cv2 \
+    IOPAINT_MODEL_DIR=/app/models \
+    IOPAINT_DEVICE=cpu \
+    TORCH_HOME=/app/models \
+    HUGGINGFACE_HUB_CACHE=/app/models \
+    IOPAINT_CACHE_DIR=/tmp/iopaint/cache
+
+# 复制项目文件
+COPY . .
+
+# 安装必要的Python依赖
+RUN pip install --no-cache-dir -e .
+
+# 安装额外的依赖，确保fastapi、uvicorn和loguru已安装
+RUN pip install --no-cache-dir fastapi uvicorn loguru opencv-python pillow numpy torch torchvision
 
+# 暴露端口
 EXPOSE 7860
 
-# 使用我们的API服务脚本
+# 启动API服务
 CMD ["python", "-m", "iopaint.api_only"]