remove extraction

app.py

@@ -54,6 +54,17 @@ _image_gen_pipeline = None
 _trellis_pipeline = None
 
 
+def start_session(req: gr.Request):
+    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+    os.makedirs(user_dir, exist_ok=True)
+    
+    
+def end_session(req: gr.Request):
+    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+    shutil.rmtree(user_dir)
+
+
+
 @spaces.GPU()
 def get_image_gen_pipeline():
     global _image_gen_pipeline
@@ -439,7 +450,7 @@ def create_interface():
             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)
+            # model_output = LitModel3D(label="Extracted GLB/Gaussian", exposure=10.0, height=300)
             
             with gr.Row():
                 download_glb = gr.DownloadButton(label="Download GLB", interactive=False)
@@ -569,6 +580,10 @@ def create_interface():
         #     outputs=[download_gs, message_box]
         # )
 
+    # Handlers
+    demo.load(start_session)
+    demo.unload(end_session)
+    
     return demo 
 
 if __name__ == "__main__":