enable zerogpu

app.py

@@ -256,6 +256,7 @@ hands = mp_hands.Hands(
     min_detection_confidence=0.1,
 )
 
+@spaces.GPU
 def make_ref_cond(
     image
 ):
@@ -263,6 +264,7 @@ def make_ref_cond(
     # print(f"image.device: {image.device}, type(image): {type(image)}")
     # image = image.to("cuda")
     print(f"autoencoder device: {next(autoencoder.parameters()).device}")
+    image = image.to(next(autoencoder.parameters()).device)
     latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...]).sample()
     return image[None, ...], latent
 
@@ -548,8 +550,7 @@ def sample_diff(ref_cond, target_cond, target_keypts, num_gen, seed, cfg):
         else:
             results.append(placeholder)
             results_pose.append(placeholder)
-    Image.fromarray(results[0]).save("sampled_image.png")
-    print("sampled_image.png saved")
+    print(f"results[0].max(): {results[0].max()}")
     return results, results_pose
 
 @spaces.GPU(duration=120)