enable zerogpu

app.py

@@ -254,7 +254,7 @@ hands = mp_hands.Hands(
     min_detection_confidence=0.1,
 )
 
-@spaces.GPU
+@spaces.GPU(duration=120)
 def get_ref_anno(ref):
     if ref is None:
         return (
@@ -367,7 +367,7 @@ def get_ref_anno(ref):
 
     return img, ref_pose, ref_cond
 
-@spaces.GPU
+@spaces.GPU(duration=120)
 def get_target_anno(target):
     if target is None:
         return (
@@ -498,7 +498,7 @@ def reset_kps(img, keypoints, side: Literal["right", "left"]):
         keypoints[1] = []
     return img, keypoints
 
-@spaces.GPU
+@spaces.GPU(duration=120)
 def sample_diff(ref_cond, target_cond, target_keypts, num_gen, seed, cfg):
     set_seed(seed)
     z = torch.randn(
@@ -541,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
+@spaces.GPU(duration=120)
 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)
@@ -696,7 +696,7 @@ def switch_mask_size(radio):
         out = (gr.update(visible=True), gr.update(visible=False))
     return out
 
-@spaces.GPU
+@spaces.GPU(duration=120)
 def sample_inpaint(
     ref_cond,
     target_cond,