Update for ZeroGPU

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.whl filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -1,61 +0,0 @@
-FROM nvidia/cuda:11.8.0-devel-ubuntu20.04
-
-ARG DEBIAN_FRONTEND=noninteractive
-
-ENV PYTHONUNBUFFERED=1
-
-ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6"
-ENV TCNN_CUDA_ARCHITECTURES=86;80;75;70;61;60
-ENV FORCE_CUDA=1
-
-ENV CUDA_HOME=/usr/local/cuda
-ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:${PATH}
-ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
-ENV LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_PATH}
-
-RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
-    build-essential \
-    curl \
-    git \
-    libegl1-mesa-dev \
-    libgl1-mesa-dev \
-    libgles2-mesa-dev \
-    libglib2.0-0 \
-    libsm6 \
-    libxext6 \
-    libxrender1 \
-    python-is-python3 \
-    python3-dev \
-    python3-pip \
-    wget \
-    && rm -rf /var/lib/apt/lists/*
-
-# Set up a new user named "user" with user ID 1000
-RUN useradd -m -u 1000 user
-# Switch to the "user" user
-USER user
-# Set home to the user's home directory
-ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH \
-    PYTHONPATH=$HOME/app \
-	PYTHONUNBUFFERED=1 \
-	GRADIO_ALLOW_FLAGGING=never \
-	GRADIO_NUM_PORTS=1 \
-	GRADIO_SERVER_NAME=0.0.0.0 \
-	GRADIO_THEME=huggingface \
-	SYSTEM=spaces
-
-RUN pip install --upgrade pip setuptools ninja
-RUN pip install torch==2.2.1 --extra-index-url https://download.pytorch.org/whl/cu118
-
-RUN python -c "import torch; print(torch.version.cuda)"
-COPY requirements.txt /tmp
-RUN cd /tmp && pip install -r requirements.txt
-
-# Set the working directory to the user's home directory
-WORKDIR $HOME/app
-
-# Copy the current directory contents into the container at $HOME/app setting the owner to the user
-COPY --chown=user . $HOME/app
-
-CMD ["python", "app.py"]

README.md

@@ -3,8 +3,9 @@ title: TripoSR
 emoji: 🐳
 colorFrom: gray
 colorTo: red
-sdk: docker
-# sdk_version: 4.19.2
+sdk: gradio
+sdk_version: 4.20.1
+python_version: 3.10.13
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -1,19 +1,23 @@
 import logging
 import os
+import shlex
+import subprocess
 import tempfile
 import time
 
 import gradio as gr
 import numpy as np
 import rembg
+import spaces
 import torch
 from PIL import Image
 from functools import partial
 
+subprocess.run(shlex.split('pip install wheel/torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl'))
+
 from tsr.system import TSR
 from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orientation
 
-HF_TOKEN = os.getenv("HF_TOKEN")
 
 HEADER = """
 # TripoSR Demo
@@ -63,7 +67,6 @@ model = TSR.from_pretrained(
     "stabilityai/TripoSR",
     config_name="config.yaml",
     weight_name="model.ckpt",
-    token=HF_TOKEN
 )
 model.renderer.set_chunk_size(131072)
 model.to(device)
@@ -95,6 +98,7 @@ def preprocess(input_image, do_remove_background, foreground_ratio):
     return image
 
 
+@spaces.GPU
 def generate(image):
     scene_codes = model(image, device=device)
     mesh = model.extract_mesh(scene_codes)[0]

requirements.txt

@@ -1,9 +1,9 @@
 omegaconf==2.3.0
 Pillow==10.1.0
 einops==0.7.0
-git+https://github.com/tatsy/torchmcubes.git
+torch==2.0.1
 transformers==4.35.0
 trimesh==4.0.5
 rembg
 huggingface-hub
-gradio
+gradio

wheel/torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4af160ba1274e2205d3529a7b82efdb6946c2158a78e19631ed840301055b8d6
+size 5824388