Update app.py

app.py

@@ -1,6 +1,28 @@
-# ... (previous imports remain the same) ...
+import gradio as gr
+from gradio_litmodel3d import LitModel3D
+
+import os
+import shutil
+os.environ['SPCONV_ALGO'] = 'native'
+from typing import *
+import torch
+import numpy as np
+import imageio
+import uuid
+from easydict import EasyDict as edict
+from PIL import Image
+from trellis.pipelines import TrellisImageTo3DPipeline
+from trellis.representations import Gaussian, MeshExtractResult
+from trellis.utils import render_utils, postprocessing_utils
+
+
+MAX_SEED = np.iinfo(np.int32).max
+TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tmp')
+os.makedirs(TMP_DIR, exist_ok=True)
+
+# Rest of the utility functions remain the same...
+[Previous utility functions: start_session, end_session, preprocess_image, pack_state, unpack_state, get_seed]
 
-@spaces.GPU
 def image_to_3d(
     image: Image.Image,
     seed: int,
@@ -107,7 +129,6 @@ def image_to_3d(
         torch.cuda.empty_cache()
         raise gr.Error(f"Processing failed: {str(e)}")
 
-@spaces.GPU
 def extract_reduced_glb(
     state: dict,
     mesh_simplify: float,
@@ -146,9 +167,9 @@ def extract_reduced_glb(
         torch.cuda.empty_cache()
         raise gr.Error(f"GLB reduction failed: {str(e)}")
 
-# ... (rest of the UI code remains the same) ...
+# Rest of the UI code and demo definition remains the same...
+[Previous UI code with Blocks, event handlers, etc.]
 
-# Add some memory optimization settings at startup
 if __name__ == "__main__":
     # Set some CUDA memory management options
     torch.cuda.empty_cache()