add GPU

app.py

@@ -19,6 +19,7 @@ import tempfile
 from typing import Dict, Tuple, Optional
 from omegaconf import OmegaConf
 import gradio as gr
+import spaces
 import numpy as np
 import trimesh
 import zero_shot
@@ -156,7 +157,7 @@ def build_outputs(surf_a: Surface, surf_b: Surface, cmap_a: np.ndarray, p2p: np.
     out_file = save_array_txt(p2p)
     return glb_a, glb_b, out_file
 
-
+@spaces.GPU
 def init_clicked(mesh1_path, mesh2_path,
                  lambda_val, zoomout_val, time_val, nloop_val, sds_val, proper_val):
     cfg.deepfeat_conf.fmap.lambda_ = lambda_val
@@ -177,7 +178,7 @@ def init_clicked(mesh1_path, mesh2_path,
     p2p_init, _ = extract_p2p_torch_fmap(C12_obj, datadicts.shape_dict["evecs"], datadicts.target_dict["evecs"])
     return build_outputs(datadicts.shape_surf, datadicts.target_surf, datadicts.cmap1, p2p_init, tag="init")
 
-
+@spaces.GPU
 def run_clicked(mesh1_path, mesh2_path, yaml_path, lambda_val, zoomout_val, time_val, nloop_val, sds_val, proper_val, progress=gr.Progress(track_tqdm=True)):
     if not mesh1_path or not mesh2_path:
         raise gr.Error("Please upload both meshes.")