gradio

gradio_app.py

@@ -133,7 +133,7 @@ class RMBG(object):
             image.save(filepath)
             return filepath
 
-# @spaces.GPU
+@spaces.GPU
 def image2mesh(image: Any, 
                more: bool = False,
                scheluder_name: str ="DDIMScheduler",
@@ -183,6 +183,7 @@ def image2mesh(image: Any,
     
     return filepath
 
+@spaces.GPU
 def mesh2texture(mesh_file, image_file):
     headers = {'Content-Type': 'application/json'}
     server_url = "114.249.238.184:34119"