enable zerogpu

app.py

@@ -20,6 +20,7 @@ from copy import deepcopy
 from typing import Optional
 import requests
 from huggingface_hub import hf_hub_download
+import spaces
 
 MAX_N = 6
 FIX_MAX_N = 6
@@ -253,7 +254,7 @@ hands = mp_hands.Hands(
     min_detection_confidence=0.1,
 )
 
-
+@spaces.GPU
 def get_ref_anno(ref):
     if ref is None:
         return (
@@ -366,7 +367,7 @@ def get_ref_anno(ref):
 
     return img, ref_pose, ref_cond
 
-
+@spaces.GPU
 def get_target_anno(target):
     if target is None:
         return (
@@ -497,7 +498,7 @@ def reset_kps(img, keypoints, side: Literal["right", "left"]):
         keypoints[1] = []
     return img, keypoints
 
-
+@spaces.GPU
 def sample_diff(ref_cond, target_cond, target_keypts, num_gen, seed, cfg):
     set_seed(seed)
     z = torch.randn(
@@ -540,7 +541,7 @@ def sample_diff(ref_cond, target_cond, target_keypts, num_gen, seed, cfg):
             results_pose.append(placeholder)
     return results, results_pose
 
-
+@spaces.GPU
 def ready_sample(img_ori, inpaint_mask, keypts):
     img = cv2.resize(img_ori[..., :3], opts.image_size, interpolation=cv2.INTER_AREA)
     sam_predictor.set_image(img)
@@ -695,7 +696,7 @@ def switch_mask_size(radio):
         out = (gr.update(visible=True), gr.update(visible=False))
     return out
 
-
+@spaces.GPU
 def sample_inpaint(
     ref_cond,
     target_cond,