ZeroGPU

demo.py

@@ -383,6 +383,7 @@ def decode_latent_to_pixel(latent_image):
     return decoded_image
 
 
+@spaces.GPU
 def check_privacy(latent_image_numpy, class_selection):
     """Check if the latent image is too similar to database images"""
     latent_image = torch.from_numpy(latent_image_numpy).to(device, dtype=dtype)