Upload Dockerfile

Dockerfile

@@ -10,10 +10,12 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     curl gcc build-essential
 
 # 创建模型和缓存目录
-RUN mkdir -p /app/models && \
-    chmod 777 /app/models && \
+RUN mkdir -p /app/models/checkpoints && \
+    chmod 777 /app/models/checkpoints && \
     mkdir -p /tmp/iopaint/models && \
-    chmod 777 /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
@@ -29,6 +31,7 @@ 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
 
 EXPOSE 7860