Update Dockerfile

Dockerfile

@@ -8,17 +8,13 @@ RUN apt-get update && apt-get install -y \
 # 安装 PyTorch
 RUN pip install torch==2.5.1 torchvision==0.20.1 --index-url https://download.pytorch.org/whl/cpu
 
-# 安装其他依赖
+# 安装其他依赖和huggingface-hub
 COPY requirements.txt .
-RUN pip install -r requirements.txt
+RUN pip install -r requirements.txt huggingface-hub
 
 # 拷贝代码
 COPY . /app
 WORKDIR /app
 
-# 创建空的模型文件（仅用于测试）
-RUN mkdir -p checkpoint && \
-    echo "这是一个空的模型文件，请替换为真实的模型" > checkpoint/final_model_K_30.pth
-
 # 入口
 CMD ["python", "app.py"]