uncomment 3

app.py

@@ -364,23 +364,23 @@ def extract_glb(
     return glb_path, glb_path
 
 
-# @spaces.GPU
-# def extract_gaussian(state: dict) -> Tuple[str, str]:
-#     """
-#     Extract a Gaussian file from the 3D model.
+@spaces.GPU
+def extract_gaussian(state: dict) -> Tuple[str, str]:
+    """
+    Extract a Gaussian file from the 3D model.
 
-#     Args:
-#         state (dict): The state of the generated 3D model.
+    Args:
+        state (dict): The state of the generated 3D model.
 
-#     Returns:
-#         str: The path to the extracted Gaussian file.
-#     """
-#     temp_dir = os.path.join(TMP_DIR, "temp_output")
-#     gs, _ = unpack_state(state)
-#     gaussian_path = os.path.join(temp_dir, 'sample.ply')
-#     gs.save_ply(gaussian_path)
-#     torch.cuda.empty_cache()
-#     return gaussian_path, gaussian_path
+    Returns:
+        str: The path to the extracted Gaussian file.
+    """
+    temp_dir = os.path.join(TMP_DIR, "temp_output")
+    gs, _ = unpack_state(state)
+    gaussian_path = os.path.join(temp_dir, 'sample.ply')
+    gs.save_ply(gaussian_path)
+    torch.cuda.empty_cache()
+    return gaussian_path, gaussian_path
 
 
 # Create a combined function that handles the whole pipeline from example to image
@@ -435,14 +435,14 @@ def create_interface():
             visual_button = gr.Button("Create visual with Flux")
 
             generated_image = gr.Image(show_label=False)
-            # gen3d_button = gr.Button("Create 3D visual with Trellis")
+            gen3d_button = gr.Button("Create 3D visual with Trellis")
 
-            # video_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300)
-            # model_output = LitModel3D(label="Extracted GLB/Gaussian", exposure=10.0, height=300)
+            video_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300)
+            model_output = LitModel3D(label="Extracted GLB/Gaussian", exposure=10.0, height=300)
             
-            # with gr.Row():
-            #     download_glb = gr.DownloadButton(label="Download GLB", interactive=False)
-            #     download_gs = gr.DownloadButton(label="Download Gaussian", interactive=False)  
+            with gr.Row():
+                download_glb = gr.DownloadButton(label="Download GLB", interactive=False)
+                download_gs = gr.DownloadButton(label="Download Gaussian", interactive=False)  
     
             message_box = gr.Textbox(
                 label="Status Messages",