Update app.py

app.py

@@ -50,12 +50,13 @@ def generate(
     num_inference_steps_refiner: int = 25,
     use_lora: bool = False,
     apply_refiner: bool = False,
-    model = 'stabilityai/stable-diffusion-xl-base-1.0',
-    vaecall = 'madebyollin/sdxl-vae-fp16-fix',
+    model = 'SG161222/Realistic_Vision_V6.0_B1_noVAE',
+    vaecall = 'stabilityai/sd-vae-ft-mse',
     lora = 'amazonaws-la/juliette',
 ) -> PIL.Image.Image:
     if torch.cuda.is_available():
-        pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
+        vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
+        pipe = DiffusionPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16)
         pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 
         if use_lora: