enable zerogpu

app.py

@@ -262,6 +262,7 @@ def make_ref_cond(
     print("ready to run autoencoder")
     # print(f"image.device: {image.device}, type(image): {type(image)}")
     # image = image.to("cuda")
+    print(f"autoencoder.encoder.device: {autoencoder.encoder.device}")
     latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...]).sample()
     return image[None, ...], latent
 
@@ -547,6 +548,8 @@ 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")
     return results, results_pose
 
 @spaces.GPU(duration=120)