Update app.py

app.py

@@ -17,6 +17,7 @@ vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype
 
 repo = "fluently/Fluently-XL-Final"
 pipe = StableDiffusionXLPipeline.from_pretrained(repo, torch_dtype=torch.float16, vae=vae)
+pipe.scheduler = EulerAncestralDiscreteScheduler(pipe.scheduler.config)
 pipe.load_lora_weights("KingNish/Better-Image-XL-Lora", weight_name="example-03.safetensors", adapter_name="lora")
 pipe.set_adapters("lora")
 pipe.to("cuda")
@@ -34,7 +35,7 @@ To optimize image results:
 """ 
 
 # Image Editor
-pipe_edit = StableDiffusionInstructPix2PixPipeline.from_pretrained( "vinesmsuic/magicbrush-jul7", torch_dtype=torch.float16, vae=vae ).to("cuda")
+pipe_edit = StableDiffusionXLInstructPix2PixPipeline.from_pretrained( "vinesmsuic/magicbrush-jul7", torch_dtype=torch.float16, vae=vae ).to("cuda")
 pipe_edit.scheduler = EulerAncestralDiscreteScheduler(pipe_edit.scheduler.config)
 
 # Generator