Update mesh.py

mesh.py

@@ -26,6 +26,7 @@ def predict_depth(model, image):
     depth = model.infer_pil(image)
     return depth
 
+@spaces.GPU(enable_queue=True)
 def get_mesh(model, image: Image.Image, keep_edges=True):
     image.thumbnail((1024,1024))  # limit the size of the input image
 
@@ -53,8 +54,7 @@ def get_mesh(model, image: Image.Image, keep_edges=True):
     mesh.export(glb_path)
     return glb_path
 
-@spaces.GPU
-def mesh_interface(model):
+def mesh_interface(model, device):
     with gr.Row():
         with gr.Column():
             inputs=[gr.Image(label="Input Image", type='pil'), gr.Checkbox(label="Keep occlusion edges", value=True)]