Upload Dockerfile with huggingface_hub

Dockerfile

@@ -4,10 +4,11 @@ FROM python:3.10-slim
 RUN useradd -m -u 1000 user
 USER root
 
-# 安装系统依赖（Blender + 渲染库 + 虚拟显示）
+# 安装系统依赖（Blender + 完整的无头渲染支持）
 RUN apt-get update && apt-get install -y \
     blender \
     xvfb \
+    xorg \
     libgl1 \
     libglib2.0-0 \
     libsm6 \
@@ -16,6 +17,8 @@ RUN apt-get update && apt-get install -y \
     libxi6 \
     libxkbcommon-x11-0 \
     libgomp1 \
+    libxxf86vm1 \
+    libxfixes3 \
     wget \
     unzip \
     git \
@@ -46,7 +49,9 @@ ENV HOME=/home/user \
     HF_HOME=/data/.huggingface \
     HF_HUB_CACHE=/data/.huggingface/hub \
     TRANSFORMERS_CACHE=/data/.huggingface/transformers \
-    HF_HUB_DISABLE_XET=1
+    HF_HUB_DISABLE_XET=1 \
+    DISPLAY=:99 \
+    LIBGL_ALWAYS_SOFTWARE=1
 
 WORKDIR $HOME/app
 
@@ -57,8 +62,18 @@ COPY --chown=user . $HOME/app
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt
 
+# 创建启动脚本（先启动 Xvfb，再运行应用）
+RUN echo '#!/bin/bash\n\
+echo "🖥️  启动虚拟显示服务器 Xvfb..."\n\
+Xvfb :99 -screen 0 1024x768x24 -ac +extension GLX +extension RENDER +render -noreset &\n\
+sleep 2\n\
+echo "✅ Xvfb 已启动 (DISPLAY=:99)"\n\
+echo "🚀 启动应用..."\n\
+exec python code/demo.py' > /home/user/start.sh && \
+    chmod +x /home/user/start.sh
+
 # 暴露端口（Gradio 默认）
 EXPOSE 7860
 
-# 启动命令
-CMD ["python", "code/demo.py"]
+# 启动命令（使用启动脚本）
+CMD ["/home/user/start.sh"]