Update app.py

app.py

@@ -13,7 +13,7 @@ from PIL import Image
 import numpy as np
 from small_256_model import UNet as small_UNet
 from big_1024_model import UNet as big_UNet
-from CLIP import load as load_clip
+from CLIP import load as load_clip,load_vae
 from rich import print as rp
 
 # Device configuration
@@ -31,8 +31,10 @@ model_repo_id = "K00B404/pix2pix_flux"
 # Global model variable
 global_model = None
 
-# CLIP
+# CLIP and VAE
 clip_model, clip_tokenizer = load_clip()
+vae = load_vae()
+
 
 def load_model():
     """Load the models at startup"""