Update app.py

app.py

@@ -59,11 +59,26 @@ pipe.to("cuda")
 
 rmbg = BiRefNet(device="cuda")
 
-@spaces.GPU(duration=600)
+@spaces.GPU(duration=120)
 def run_rembg(image: Image.Image):
     return rmbg.run(image)
+
+@spaces.GPU(duration=120)
+def generate_mesh(
+    image: Any, 
+    resolution: str = '1024', 
+):
+    torch.cuda.empty_cache()
+    mesh = pipe(
+        image, 
+        sdf_resolution=int(resolution), 
+        mc_threshold=0.2,
+        remesh=False,
+    )["mesh"]
+    torch.cuda.empty_cache()
+    return mesh
+
     
-@spaces.GPU(duration=600)
 def image2mesh(
     image: Any, 
     resolution: str = '1024', 
@@ -71,29 +86,24 @@ def image2mesh(
     simplify_ratio: float = 0.95, 
     output_path: str = 'outputs/web'
 ):
-    
-    torch.cuda.empty_cache()
-    
+        
     if not os.path.exists(output_path):
         os.makedirs(output_path)
     
     uid = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
     image.save(os.path.join(output_path, uid + '.png'))
 
-    mesh = pipe(
-        image, 
-        sdf_resolution=int(resolution), 
-        mc_threshold=0.2,
-        remesh=simplify,
-        simplify_ratio=simplify_ratio,
-    )["mesh"]
+    mesh = generate_mesh(image, resolution)
+
+    if simplify:
+        mesh = postprocess_mesh(mesh.vertices, mesh.faces, simplify=simplify, 
+                                simplify_ratio=simplify_ratio, fill_holes=False, verbose=True)
 
     mesh_path = os.path.join(output_path, f'{uid}.obj')
     mesh.export(
         mesh_path,
         include_normals=True,
     )
-    torch.cuda.empty_cache()
     
     return mesh_path