Update app.py

Key changes:

Added new extract_high_quality_mesh function that:

Just extracts vertices and faces
Skips UV mapping and texture baking
Creates simple untextured GLB


Updated UI text to clarify that full-quality version is untextured
Maintained original texture functionality for reduced version
Added proper memory cleanup throughout

This should now avoid the GPU timeout since we're not doing the heavy texture baking for the full-quality version. Let me know if you'd like any adjustments!

app.py

@@ -10,6 +10,7 @@ import torch
 import numpy as np
 import imageio
 import uuid
+import trimesh
 from easydict import EasyDict as edict
 from PIL import Image
 from trellis.pipelines import TrellisImageTo3DPipeline
@@ -161,32 +162,27 @@ def image_to_3d(
     return state, video_path
 
 @spaces.GPU
-def extract_high_quality_glb(
+def extract_high_quality_mesh(
     state: dict,
     req: gr.Request,
 ) -> Tuple[str, str]:
     """
-    Extract a high-quality GLB file with memory optimization.
+    Extract just the high-quality mesh without texturing.
     """
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     gs, mesh, trial_id = unpack_state(state)
     
-    # Clear cache before GLB generation
+    # Clear cache before starting
     torch.cuda.empty_cache()
     
-    # Process mesh in original quality (no reduction)
-    glb = postprocessing_utils.to_glb(
-        gs,
-        mesh,
-        simplify=0.0,  # No simplification
-        fill_holes=True,
-        fill_holes_max_size=0.04,
-        texture_size=2048,  # Maximum texture resolution
-        verbose=True  # Show progress
-    )
+    # Get just the mesh vertices and faces at full quality
+    vertices = mesh.vertices.cpu().numpy()
+    faces = mesh.faces.cpu().numpy()
     
+    # Create a simple untextured GLB
+    simple_mesh = trimesh.Trimesh(vertices=vertices, faces=faces)
     glb_path = os.path.join(user_dir, f"{trial_id}_full.glb")
-    glb.export(glb_path)
+    simple_mesh.export(glb_path)
     
     # Final cleanup
     torch.cuda.empty_cache()
@@ -200,7 +196,7 @@ def extract_reduced_glb(
     req: gr.Request,
 ) -> Tuple[str, str]:
     """
-    Extract a reduced-quality GLB file with memory optimization.
+    Extract a reduced-quality GLB file with texturing.
     """
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     gs, mesh, trial_id = unpack_state(state)
@@ -227,8 +223,8 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
     ## Image to 3D Asset with [TRELLIS](https://trellis3d.github.io/)
     * Upload an image and click "Generate" to create a 3D asset
     * After generation:
-        * Click "Extract Full GLB" for maximum quality (no reduction)
-        * Or use GLB Extraction Settings for a reduced version
+        * Click "Extract Full GLB" for maximum quality mesh (untextured)
+        * Or use GLB Extraction Settings for a reduced textured version
     """)
     
     with gr.Row():
@@ -303,7 +299,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
     )
 
     extract_full_btn.click(
-        extract_high_quality_glb,
+        extract_high_quality_mesh,
         inputs=[output_buf],
         outputs=[model_output, download_full],
     ).then(