Update app.py

app.py

@@ -100,6 +100,7 @@ else:
 model_cache_dir = './ckpts/'
 os.makedirs(model_cache_dir, exist_ok=True)
 
+@spaces.GPU
 def get_render_cameras(batch_size=1, M=120, radius=4.0, elevation=20.0, is_flexicubes=False, fov=50):
     """
     Get the rendering camera parameters.
@@ -169,7 +170,7 @@ def get_render_cameras(batch_size=1, M=120, radius=4.0, elevation=20.0, is_flexi
     all_campos = torch.stack(all_campos, dim=0).unsqueeze(0).squeeze(2)
     return all_mv, all_mvp, all_campos
 
-
+@spaces.GPU
 def render_frames(model, planes, render_cameras, camera_pos, env, materials, render_size=512, chunk_size=1, is_flexicubes=False):
     """
     Render frames from triplanes.
@@ -305,7 +306,7 @@ def preprocess(input_image, do_remove_background):
 
     return input_image
 
-
+@spaces.GPU
 def generate_mvs(input_image, sample_steps, sample_seed):
 
     seed_everything(sample_seed)
@@ -352,7 +353,7 @@ def make_mesh(mesh_fpath, planes):
 
     return mesh_fpath, mesh_glb_fpath
 
-
+@spaces.GPU
 def make3d(images):
 
     images = np.asarray(images, dtype=np.float32) / 255.0