Xiaomeng1130 commited on
Commit
47f52c4
·
verified ·
1 Parent(s): 607a3d1

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +12 -13
Dockerfile CHANGED
@@ -5,26 +5,26 @@ FROM python:3.10-slim
5
  WORKDIR /app
6
 
7
  # 安装必要的系统依赖 (如 git-lfs)
 
8
  RUN apt-get update && apt-get install -y \
9
-     git \
10
-     git-lfs \
11
-     ffmpeg \
12
-     libsm6 \
13
-     libxext6 \
14
-     cmake \
15
-     rsync \
16
-     libgl1 \
17
-     && rm -rf /var/lib/apt/lists/*
18
 
19
  # 复制所有文件到容器中
20
  COPY . /app
21
 
22
  # 使用 pip 安装依赖,包括本地包安装
23
- # 这会安装所有依赖,包括您在 requirements.txt 中添加的 -e .
24
  RUN pip install --no-cache-dir -r requirements.txt
25
 
26
  # =======================================================
27
- # 🚀 解决超时问题:预下载模型 (新增步骤)
28
  # =======================================================
29
 
30
  # 1. 安装 Hugging Face CLI,用于管理模型下载
@@ -32,11 +32,10 @@ RUN pip install huggingface-hub
32
 
33
  # 2. 预下载模型权重到构建缓存中
34
  # 请将 'Xiaomeng1130/stoma-clip-model' 替换为您的 Stoma-CLIP 模型在 Hugging Face 上的实际仓库名
35
- # 这样模型文件会被下载到 /root/.cache,应用启动时直接读取,无需再次下载
36
  RUN huggingface-cli download --repo-id Xiaomeng1130/stoma-clip-model --cache-dir /root/.cache/huggingface/hub
37
 
38
  # =======================================================
39
  # 启动应用
40
  # =======================================================
41
  # 运行应用(Gradion 默认运行在 7860 端口)
42
- CMD ["python", "app.py"]
 
5
  WORKDIR /app
6
 
7
  # 安装必要的系统依赖 (如 git-lfs)
8
+ # **修正:移除了导致错误的不可见字符和多余的空格。**
9
  RUN apt-get update && apt-get install -y \
10
+ git \
11
+ git-lfs \
12
+ ffmpeg \
13
+ libsm6 \
14
+ libxext6 \
15
+ cmake \
16
+ rsync \
17
+ libgl1 \
18
+ && rm -rf /var/lib/apt/lists/*
19
 
20
  # 复制所有文件到容器中
21
  COPY . /app
22
 
23
  # 使用 pip 安装依赖,包括本地包安装
 
24
  RUN pip install --no-cache-dir -r requirements.txt
25
 
26
  # =======================================================
27
+ # 🚀 解决超时问题:预下载模型 (保持不变)
28
  # =======================================================
29
 
30
  # 1. 安装 Hugging Face CLI,用于管理模型下载
 
32
 
33
  # 2. 预下载模型权重到构建缓存中
34
  # 请将 'Xiaomeng1130/stoma-clip-model' 替换为您的 Stoma-CLIP 模型在 Hugging Face 上的实际仓库名
 
35
  RUN huggingface-cli download --repo-id Xiaomeng1130/stoma-clip-model --cache-dir /root/.cache/huggingface/hub
36
 
37
  # =======================================================
38
  # 启动应用
39
  # =======================================================
40
  # 运行应用(Gradion 默认运行在 7860 端口)
41
+ CMD ["python", "app.py"]