Update Dockerfile

Dockerfile

@@ -6,23 +6,37 @@ WORKDIR /app
 
 # 安装必要的系统依赖 (如 git-lfs)
 RUN apt-get update && apt-get install -y \
-    git \
-    git-lfs \
-    ffmpeg \
-    libsm6 \
-    libxext6 \
-    cmake \
-    rsync \
-    libgl1 \
-    && rm -rf /var/lib/apt/lists/*
+    git \
+    git-lfs \
+    ffmpeg \
+    libsm6 \
+    libxext6 \
+    cmake \
+    rsync \
+    libgl1 \
+    && rm -rf /var/lib/apt/lists/*
 
 # 复制所有文件到容器中
-# 确保 setup.py 和 requirements.txt 都在这里
 COPY . /app
 
 # 使用 pip 安装依赖，包括本地包安装
-# 注意：我们现在明确执行安装，它不会再抱怨 setup.py 找不到
+# 这会安装所有依赖，包括您在 requirements.txt 中添加的 -e .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# =======================================================
+# 🚀 解决超时问题：预下载模型 (新增步骤)
+# =======================================================
+
+# 1. 安装 Hugging Face CLI，用于管理模型下载
+RUN pip install huggingface-hub
+
+# 2. 预下载模型权重到构建缓存中
+# 请将 'Xiaomeng1130/stoma-clip-model' 替换为您的 Stoma-CLIP 模型在 Hugging Face 上的实际仓库名
+# 这样模型文件会被下载到 /root/.cache，应用启动时直接读取，无需再次下载
+RUN huggingface-cli download --repo-id Xiaomeng1130/stoma-clip-model --cache-dir /root/.cache/huggingface/hub
+
+# =======================================================
+# 启动应用
+# =======================================================
 # 运行应用（Gradion 默认运行在 7860 端口）
 CMD ["python", "app.py"]