Update app.py

app.py

@@ -14,16 +14,27 @@ from PIL import Image
 from trellis.pipelines import TrellisImageTo3DPipeline
 from trellis.representations import Gaussian, MeshExtractResult
 from trellis.utils import render_utils, postprocessing_utils
+import gradio as gr
+from gradio_client import Client, handle_file
+from gradio_client.exceptions import AppError
+from gradio_litmodel3d import LitModel3D
+
+import tempfile
+import trimesh
+import time
 
 
 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)
 
+RIGNET_SPACE = "ckc99u/spaceB"  # Your RigNet space
+HF_TOKEN = os.getenv("HF_TOKEN")
 
 def start_session(req: gr.Request):
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     os.makedirs(user_dir, exist_ok=True)
+    os.makedirs(user_dir, exist_ok=True)
     
     
 def end_session(req: gr.Request):