try to figure out how ZeroGPU works

app.py

@@ -21,10 +21,11 @@ from Utility.storage_config import MODELS_DIR
 
 class ControllableInterface(torch.nn.Module):
 
+    @spaces.GPU
     def __init__(self, available_artificial_voices=1000):
         super().__init__()
         self.model = ToucanTTSInterface(device="cpu", tts_model_path="Meta")
-        self.wgan = GanWrapper(os.path.join(MODELS_DIR, "Embedding", "embedding_gan.pt"), device="cpu")
+        self.wgan = GanWrapper(os.path.join(MODELS_DIR, "Embedding", "embedding_gan.pt"), device="cuda")
         self.generated_speaker_embeds = list()
         self.available_artificial_voices = available_artificial_voices
         self.current_language = ""