model3d component

app.py

@@ -189,21 +189,16 @@ def process(input_image, prompt, prompt_neg='', input_elevation=0, input_num_ste
         images = np.concatenate(images, axis=0)
         imageio.mimwrite(output_video_path, images, fps=30)
 
-    return mv_image_grid, output_video_path, output_ply_path
+    return output_ply_path, output_ply_path
 
 # gradio UI
 
-_TITLE = '''LGM: Large Multi-View Gaussian Model for High-Resolution 3D Content Creation'''
+_TITLE = '''LGM Mini'''
 
 _DESCRIPTION = '''
 <div>
-<a style="display:inline-block" href="https://me.kiui.moe/lgm/"><img src='https://img.shields.io/badge/public_website-8A2BE2'></a>
-<a style="display:inline-block; margin-left: .5em" href="https://github.com/3DTopia/LGM"><img src='https://img.shields.io/github/stars/3DTopia/LGM?style=social'/></a>
+A lightweight version of <a href="https://huggingface.co/spaces/ashawkey/LGM">LGM: Large Multi-View Gaussian Model for High-Resolution 3D Content Creation</a>
 </div>
-
-* Input can be only text, only image, or both image and text. 
-* Output is a `ply` file containing the 3D Gaussians, please check our [repo](https://github.com/3DTopia/LGM/blob/main/readme.md) for visualization and mesh conversion.
-* If you find the output unsatisfying, try using different seeds!
 '''
 
 block = gr.Blocks(title=_TITLE).queue()
@@ -232,16 +227,10 @@ with block:
 
         
         with gr.Column(scale=1):
-            with gr.Tab("Video"):
-                # final video results
-                output_video = gr.Video(label="video")
-                # ply file
-                output_file = gr.File(label="3D Gaussians (ply format)")
-            with gr.Tab("Multi-view Image"):
-                # multi-view results
-                output_image = gr.Image(interactive=False, show_label=False)
-
-        button_gen.click(process, inputs=[input_image, input_text, input_neg_text, input_elevation, input_num_steps, input_seed], outputs=[output_image, output_video, output_file])
+            output_splat = gr.Model3D(label="3D Gaussians")
+            output_file = gr.File(label="3D Gaussians (ply format)")
+
+        button_gen.click(process, inputs=[input_image, input_text, input_neg_text, input_elevation, input_num_steps, input_seed], outputs=[output_splat, output_file])
     
     gr.Examples(
         examples=[
@@ -253,7 +242,7 @@ with block:
             "data_test/gso_rabbit.jpg",
         ],
         inputs=[input_image],
-        outputs=[output_image, output_video, output_file],
+        outputs=[output_splat, output_file],
         fn=lambda x: process(input_image=x, prompt=''),
         cache_examples=True,
         label='Image-to-3D Examples'
@@ -270,7 +259,7 @@ with block:
 
         ],
         inputs=[input_text],
-        outputs=[output_image, output_video, output_file],
+        outputs=[output_splat, output_file],
         fn=lambda x: process(input_image=None, prompt=x),
         cache_examples=True,
         label='Text-to-3D Examples'