Update app.py

app.py

@@ -2,6 +2,11 @@ import gradio as gr
 import spaces
 from gradio_litmodel3d import LitModel3D
 
+import os
+import shutil
+import subprocess
+import sys
+
 # Install local wheels at runtime
 def install_local_wheels():
     """Install the local wheel files that couldn't be installed during Docker build."""
@@ -19,8 +24,6 @@ def install_local_wheels():
 # Install wheels before importing trellis
 install_local_wheels()
 
-import os
-import shutil
 os.environ['SPCONV_ALGO'] = 'native'
 from typing import *
 import torch
@@ -46,7 +49,8 @@ def start_session(req: gr.Request):
     
 def end_session(req: gr.Request):
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
-    shutil.rmtree(user_dir)
+    if os.path.exists(user_dir):
+        shutil.rmtree(user_dir)
 
 
 def preprocess_image(image: Image.Image) -> Image.Image: