Update app.py

app.py

@@ -18,7 +18,9 @@ class dotdict(dict):
     __setattr__ = dict.__setitem__
     __delattr__ = dict.__delitem__
 
+
 class InferRunner:
+    @spaces.GPU()
     def __init__(self, device):
         vae_config = json.load(open("ckpts/ldm/vae_config.json"))
         self.vae = AutoencoderKL(**vae_config).to(device)