hf cuda issue

app.py

@@ -282,6 +282,7 @@ def prepare_anno(ref, ref_is_user):
     else:
         return img, None
 
+@spaces_60_fn
 def get_ref_anno(img, keypts, use_mask, use_pose):
     no_mask, no_pose = not use_mask, not use_pose
     if img.sum() == 0:  # clear_all
@@ -313,6 +314,7 @@ def get_ref_anno(img, keypts, use_mask, use_pose):
         hand_mask = np.zeros_like(img[:,:, 0])
         ref_pose = visualize_hand(keypts, img)
     else:
+        sam_predictor = init_sam(ckpt_path=sam_path, device="cuda")
         sam_predictor.set_image(img)
         if keypts[0].sum() != 0 and keypts[21].sum() != 0:
             # input_point = np.array([keypts[0], keypts[21]])