Adapt to ZeroGPU?

app.py

@@ -43,6 +43,7 @@ def _safe_ext(path: str) -> str:
             return ext
     return os.path.splitext(path)[1].lower()
 
+@spaces.GPU
 def convert_and_show(mesh_file):
     os.makedirs("tmp/glbs", exist_ok=True)
     if mesh_file is None:
@@ -53,6 +54,7 @@ def convert_and_show(mesh_file):
     mesh.export(f_name)
     return f_name
 
+@spaces.GPU
 def convert_and_show_twice(mesh_file_1, mesh_file_2):
     return convert_and_show(mesh_file_1), convert_and_show(mesh_file_2)
 
@@ -184,9 +186,9 @@ def init_clicked(mesh1_path, mesh2_path,
     matcher.reconf(cfg)
     if not mesh1_path or not mesh2_path:
         raise gr.Error("Please upload both meshes.")
-    matcher._init()
     global datadicts
     datadicts = Datadicts(mesh1_path, mesh2_path)
+    matcher._init()
     shape_dict, target_dict = convert_dict(datadicts.shape_dict, 'cuda'), convert_dict(datadicts.target_dict, 'cuda')
     fmap_model_cuda = matcher.fmap_model.cuda()
     diff_model_cuda = matcher.diffusion_model

zero_shot.py

@@ -66,7 +66,7 @@ class Tee:
 
 class DiffModel:
 
-    def __init__(self, cfg, device="cuda:0"):
+    def __init__(self, cfg, device="cpu"):
         if cfg["train_dir"] == "pretrained":
             url = "https://huggingface.co/daidedou/diffumatch_model/resolve/main/network-snapshot-041216.pkl"
             network_pkl = ensure_pretrained_file(url, "pretrained")