update

README.md

@@ -4,7 +4,7 @@ emoji: 🖼️
 colorFrom: blue
 colorTo: red
 sdk: gradio
-sdk_version: 5.12.0
+sdk_version: 6.8.0
 app_file: app.py
 pinned: false
 license: apache-2.0

app.py

@@ -19,7 +19,7 @@ import os
 
 # GRADIO_APP == "imageto3d_sam3d", sam3d object model, by default.
 # GRADIO_APP == "imageto3d", TRELLIS model.
-os.environ["GRADIO_APP"] = "imageto3d"
+os.environ["GRADIO_APP"] = "imageto3d_sam3d"
 from glob import glob
 
 import gradio as gr

common.py

@@ -16,10 +16,8 @@
 
 import spaces
 from embodied_gen.utils.monkey_patch.trellis import monkey_path_trellis
-
 monkey_path_trellis()
-from embodied_gen.utils.monkey_patch.gradio import _patch_gradio_schema_bool_bug, _patch_open3d_cuda_device_count_bug
-_patch_gradio_schema_bool_bug()
+from embodied_gen.utils.monkey_patch.gradio import _patch_open3d_cuda_device_count_bug
 _patch_open3d_cuda_device_count_bug()
 
 import gc

embodied_gen/models/sam3d.py

@@ -22,7 +22,6 @@ import sys
 
 import numpy as np
 from hydra.utils import instantiate
-# from modelscope import snapshot_download
 from huggingface_hub import snapshot_download
 from omegaconf import OmegaConf
 from PIL import Image
@@ -67,7 +66,6 @@ class Sam3dInference:
         self, local_dir: str = "weights/sam-3d-objects", compile: bool = False, device: str = "cuda",
     ) -> None:
         if not os.path.exists(local_dir):
-            # snapshot_download("facebook/sam-3d-objects", local_dir=local_dir)
             snapshot_download("tuandao-zenai/sam-3d-objects", local_dir=local_dir)
         config_file = os.path.join(local_dir, "checkpoints/pipeline.yaml")
         config = OmegaConf.load(config_file)

embodied_gen/utils/monkey_patch/gradio.py

@@ -18,10 +18,9 @@
 import gradio_client.utils as gradio_client_utils
 import fileinput
 import site
-from pathlib import Path
 
 def _patch_gradio_schema_bool_bug() -> None:
-    """Patch gradio_client schema parser for bool-style additionalProperties."""
+    """Patch schema parser for bool-style for gradio<5.33."""
     original_get_type = gradio_client_utils.get_type
     original_json_schema_to_python_type = (
         gradio_client_utils._json_schema_to_python_type

requirements.txt

@@ -20,7 +20,8 @@ igraph==0.11.8
 pyvista==0.36.1
 openai==1.58.1
 transformers==4.42.4
-gradio==5.12.0
+gradio==6.8.0
+spaces==0.47.0
 sentencepiece==0.2.0
 diffusers==0.31.0
 xatlas==0.0.9
@@ -31,7 +32,6 @@ basicsr==1.4.2
 realesrgan==0.3.0
 pydantic
 vtk==9.3.1
-spaces==0.47.0
 # utils3d@git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
 clip@git+https://github.com/openai/CLIP.git
 kolors@git+https://github.com/Kwai-Kolors/Kolors.git#egg=038818d