Hanxiaofeng123 commited on
Commit
e51d2e3
·
verified ·
1 Parent(s): 2fb8f68

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -17
Dockerfile CHANGED
@@ -3,7 +3,6 @@ FROM python:3.10-slim
3
  # 安装系统依赖
4
  RUN apt-get update && apt-get install -y \
5
  build-essential \
6
- wget \
7
  && rm -rf /var/lib/apt/lists/*
8
 
9
  # 安装 PyTorch
@@ -17,20 +16,9 @@ RUN pip install -r requirements.txt
17
  COPY . /app
18
  WORKDIR /app
19
 
20
- # 创建 checkpoint 目录
21
- RUN mkdir -p checkpoint
 
22
 
23
- # 下载模型文件的脚本
24
- RUN echo '#!/bin/sh\n\
25
- if [ ! -f "checkpoint/final_model_K_30.pth" ]; then\n\
26
- echo "下载模型文件..."\n\
27
- wget -O checkpoint/final_model_K_30.pth \\\n\
28
- "https://huggingface.co/spaces/Hanxiaofeng123/Deepcube/resolve/main/checkpoint/final_model_K_30.pth"\n\
29
- else\n\
30
- echo "模型文件已存在"\n\
31
- ls -lh checkpoint/final_model_K_30.pth\n\
32
- fi' > /download_model.sh && \
33
- chmod +x /download_model.sh
34
-
35
- # 入口(先下载模型,再运行应用)
36
- CMD ["sh", "-c", "/download_model.sh && python app.py"]
 
3
  # 安装系统依赖
4
  RUN apt-get update && apt-get install -y \
5
  build-essential \
 
6
  && rm -rf /var/lib/apt/lists/*
7
 
8
  # 安装 PyTorch
 
16
  COPY . /app
17
  WORKDIR /app
18
 
19
+ # 创建空的模型文件(仅用于测试)
20
+ RUN mkdir -p checkpoint && \
21
+ echo "这是一个空的模型文件,请替换为真实的模型" > checkpoint/final_model_K_30.pth
22
 
23
+ # 入口
24
+ CMD ["python", "app.py"]