Update app.py

app.py

@@ -13,7 +13,7 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 DTYPE = torch.float16
 
 def load_vae(model_id=MODEL_ID, device=DEVICE):
-    for attempt in (None, "vae4"):
+    for attempt in (None, "vae5"):
         try:
             if attempt is None:
                 vae = AsymmetricAutoencoderKL.from_pretrained(model_id, torch_dtype=DTYPE)