update

app.py

@@ -18,6 +18,7 @@ from common import (
 )
 from gradio.themes import Default
 from gradio.themes.utils.colors import slate
+from gradio_litmodel3d import LitModel3D
 
 with gr.Blocks(
     delete_cache=(43200, 43200), theme=Default(primary_hue=slate)
@@ -217,8 +218,8 @@ with gr.Blocks(
                 loop=True,
                 height=300,
             )
-            model_output_gs = gr.Model3D(
-                label="Gaussian Representation", height=300, interactive=False
+            model_output_gs = LitModel3D(
+                label="Gaussian Representation", height=300, interactive=False, exposure=10,
             )
             aligned_gs = gr.Textbox(visible=False)
             with gr.Row():
@@ -427,4 +428,4 @@ with gr.Blocks(
 
 
 if __name__ == "__main__":
-    demo.launch(server_name="10.34.8.82", server_port=8084)
+    demo.launch()

requirements.txt

@@ -19,8 +19,8 @@ igraph==0.11.8
 pyvista==0.36.1
 openai==1.58.1
 transformers==4.42.4
-# gradio_litmodel3d==0.0.1
-gradio==5.12.0
+gradio_litmodel3d==0.0.1
+# gradio==5.12.0
 sentencepiece==0.2.0
 diffusers==0.31.0
 xatlas==0.0.9
@@ -38,10 +38,10 @@ kolors@git+https://github.com/Kwai-Kolors/Kolors.git#egg=038818d
 segment-anything@git+https://github.com/facebookresearch/segment-anything.git#egg=dca509f
 https://github.com/nerfstudio-project/gsplat/releases/download/v1.5.0/gsplat-1.5.0+pt24cu121-cp310-cp310-linux_x86_64.whl
 https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.0.post2/flash_attn-2.7.0.post2+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
-# https://huggingface.co/spaces/xinjjj/ImgRoboAssetGen/blob/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
-# https://huggingface.co/spaces/xinjjj/ImgRoboAssetGen/blob/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl
-# https://huggingface.co/spaces/xinjjj/ImgRoboAssetGen/blob/main/wheels/kaolin-0.16.0-cp310-cp310-linux_x86_64.whl
+https://huggingface.co/spaces/xinjjj/ImgRoboAssetGen/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
+https://huggingface.co/spaces/xinjjj/ImgRoboAssetGen/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl
+https://huggingface.co/spaces/xinjjj/ImgRoboAssetGen/resolve/main/wheels/kaolin-0.16.0-cp310-cp310-linux_x86_64.whl
 
-https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
-https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl
-kaolin@git+https://github.com/NVIDIAGameWorks/[email protected]
+# https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
+# https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl
+# kaolin@git+https://github.com/NVIDIAGameWorks/[email protected]