🎉 init

Dockerfile

@@ -0,0 +1,33 @@
+FROM python:3.10
+
+RUN apt-get update -y && \
+    apt-get install -y libegl1-mesa-dev libgl1-mesa-dev libxi6 libxkbcommon-x11-0 libxfixes3 && \
+    apt-get clean && \
+    apt-get autoremove
+
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+
+WORKDIR $HOME
+
+RUN git clone https://github.com/jasongzy/Puppeteer --recursive app
+
+WORKDIR $HOME/app
+
+RUN python -m pip install --upgrade pip && \
+    pip install torch==2.1.1 torchvision==0.16.1 --index-url https://download.pytorch.org/whl/cu118 && \
+    pip install cython==0.29.36 && \
+    pip install tetgen==0.5.2 --no-build-isolation && \
+    pip install -r requirements.txt && \
+    if python -c "import torch; exit(0 if torch.cuda.is_available() else 1)" 2>/dev/null; then \
+        pip install flash-attn==2.6.3 --no-build-isolation --no-cache-dir; \
+    fi && \
+    pip cache purge
+
+# RUN python skeleton/download.py && \
+#     python skinning/download.py && \
+#     python animation/download.py
+
+CMD ["python", "app.py"]

README.md

@@ -1,11 +1,13 @@
 ---
 title: Puppeteer
-emoji: 🐢
-colorFrom: blue
-colorTo: purple
+emoji: 🤖
+colorFrom: gray
+colorTo: red
 sdk: docker
 pinned: false
 short_description: Rig and Animate Your 3D Models
+models:
+  - Seed3D/Puppeteer
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference