Update app.py

app.py

@@ -144,10 +144,10 @@ controlnet_canny = ControlNetModel.from_pretrained(
     torch_dtype=torch.float16
 )
 # when test with other base model, you need to change the vae also.
-vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
+vae = AutoencoderKL.from_pretrained("John6666/pony-realism-v21main-sdxl", subfolder="vae", torch_dtype=torch.float16)
 
 pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-xl-base-1.0",
+    "John6666/pony-realism-v21main-sdxl",
     controlnet=controlnet,
     vae=vae,
     torch_dtype=torch.float16,
@@ -156,14 +156,13 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
 pipe.to(device)
 # Load model.
 pipe_canny = StableDiffusionXLControlNetPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-xl-base-1.0",
+    "John6666/pony-realism-v21main-sdxl",
     controlnet=controlnet_canny,
     vae=vae,
     safety_checker=None,
     torch_dtype=torch.float16,
     scheduler=eulera_scheduler,
 )
-
 pipe_canny.to(device)
 
 MAX_SEED = np.iinfo(np.int32).max