Update app.py

app.py

@@ -21,8 +21,8 @@ class dotdict(dict):
 
 class InferRunner:
     def __init__(self, device):
-        self.vae = AutoencoderKL(**vae_config)
         vae_config = json.load(open("ckpts/ldm/vae_config.json"))
+        self.vae = AutoencoderKL(**vae_config)
         vae_weights = torch.load("ckpts/ldm/pytorch_model_vae.bin", map_location="cpu")
         self.vae.load_state_dict(vae_weights)
         self.vae = self.vae.to(device)