Update Dockerfile

Dockerfile

@@ -1,18 +1,11 @@
-FROM python:3.10  # 使用完整版本，不是 slim
+FROM python:3.10-slim
 
 # 安装系统依赖
 RUN apt-get update && apt-get install -y \
     build-essential \
-    curl \
-    git \
-    file \
+    wget \
     && rm -rf /var/lib/apt/lists/*
 
-# 安装 Git LFS
-RUN curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | bash && \
-    apt-get install -y git-lfs && \
-    git lfs install
-
 # 安装 PyTorch
 RUN pip install torch==2.5.1 torchvision==0.20.1 --index-url https://download.pytorch.org/whl/cpu
 
@@ -24,15 +17,20 @@ RUN pip install -r requirements.txt
 COPY . /app
 WORKDIR /app
 
-# 初始化 Git 仓库并拉取 LFS 文件
-RUN git init && \
-    git lfs install --local && \
-    git add . && \
-    git lfs pull
-
-# 验证文件
-RUN ls -la checkpoint/ && \
-    ls -lh checkpoint/final_model_K_30.pth
-
-# 入口
-CMD ["python", "app.py"]
+# 创建 checkpoint 目录
+RUN mkdir -p checkpoint
+
+# 下载模型文件的脚本
+RUN echo '#!/bin/sh\n\
+if [ ! -f "checkpoint/final_model_K_30.pth" ]; then\n\
+    echo "下载模型文件..."\n\
+    wget -O checkpoint/final_model_K_30.pth \\\n\
+    "https://huggingface.co/spaces/Hanxiaofeng123/Deepcube/blob/main/checkpoint/final_model_K_30.pth"\n\
+else\n\
+    echo "模型文件已存在"\n\
+    ls -lh checkpoint/final_model_K_30.pth\n\
+fi' > /download_model.sh && \
+    chmod +x /download_model.sh
+
+# 入口（先下载模型，再运行应用）
+CMD ["sh", "-c", "/download_model.sh && python app.py"]