enable zerogpu

app.py

@@ -256,6 +256,15 @@ hands = mp_hands.Hands(
     min_detection_confidence=0.1,
 )
 
+@spaces.GPU(duration=60)
+def make_ref_cond(
+    image
+):
+    print("ready to run autoencoder")
+    # print(f"image.device: {image.device}, type(image): {type(image)}")
+    latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...].to("cuda")).sample()
+    return image[None, ...], latent
+
 def get_ref_anno(ref):
     print("inside get_ref_anno")
     if ref is None:
@@ -322,15 +331,6 @@ def get_ref_anno(ref):
         hand_mask = np.zeros_like(img[:,:, 0])
         ref_pose = np.zeros_like(img)
 
-    @spaces.GPU(duration=60)
-    def make_ref_cond(
-        image
-    ):
-        print("ready to run autoencoder")
-        # print(f"image.device: {image.device}, type(image): {type(image)}")
-        latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...]).sample()
-        return image[None, ...], latent
-
     image_transform = Compose(
         [
             ToTensor(),