Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- 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 |
-
#
|
| 21 |
-
RUN mkdir -p checkpoint
|
|
|
|
| 22 |
|
| 23 |
-
#
|
| 24 |
-
|
| 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"]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|