Update Dockerfile

Dockerfile

@@ -1,28 +1,35 @@
 FROM python:3.10-slim
 
-# 安装系统依赖
+# 安装系统依赖，包括 git 和 git-lfs
 RUN apt-get update && apt-get install -y \
     build-essential \
+    git \
     && rm -rf /var/lib/apt/lists/*
 
-# 安装 PyTorch（CPU 版本即可，避免太大）
+# 安装 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
 
 # 安装其他依赖
 COPY requirements.txt .
 RUN pip install -r requirements.txt
 
-# 拷贝代码
-COPY . /app
+# 克隆仓库（包含 LFS 文件）
+RUN git clone https://huggingface.co/你的用户名/你的仓库 /app
+
 WORKDIR /app
 
-# 添加验证命令
-RUN echo "检查文件是否存在：" && \
-    ls -la /app/checkpoint/ && \
-    echo "查找 .pth 文件：" && \
-    find /app -name "*.pth" -o -name "*.PTH" && \
-    echo "当前目录结构：" && \
-    find /app -type f -name "*.pth" | head -10
+# 拉取 LFS 文件
+RUN git lfs pull
+
+# 验证文件
+RUN ls -la checkpoint/ && \
+    file checkpoint/final_model_K_30.pth && \
+    du -h checkpoint/final_model_K_30.pth
 
 # 入口
-CMD ["python", "app.py"]
+CMD ["python", "app.py"]