Update app.py

app.py

@@ -8,12 +8,12 @@ import io
 import tempfile
 import os
 
-MODEL_ID = "AiArtLab/simplevae2x"
+MODEL_ID = "babkasotona/vae8x16x32ch"
 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, "vae5"):
+    for attempt in (None, "vae"):
         try:
             if attempt is None:
                 vae = AsymmetricAutoencoderKL.from_pretrained(model_id, torch_dtype=DTYPE)
@@ -62,8 +62,7 @@ def encode_decode(img: Image.Image):
 with gr.Blocks(title="Asymmetric VAE 2x Upscaler") as demo:
     gr.Markdown("""
     # 🧠 Asymmetric VAE 2x Upscaler  
-    Тестовое демо модели [`AiArtLab/simplevae2x`](https://huggingface.co/AiArtLab/simplevae2x).  
-    Загрузите изображение → нажмите **"Upscale"** → сохраните результат.
+    Загрузите изображение → нажмите **"Upscale"** 
     """)
 
     with gr.Column():