change model

app.py

@@ -13,8 +13,8 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
-# Load the initial VAE model for preprocessing YAY
-vae_model_name = "CompVis/stable-diffusion-v1-4"  # Example VAE model
+# Load the initial VAE model for preprocessing
+vae_model_name = "runwayml/stable-diffusion-v1-5"  # Changed VAE model
 vae = AutoencoderKL.from_pretrained(vae_model_name).to(device)
 
 # Load the FLUX diffusion pipeline with optimizations