Update app.py

app.py

@@ -147,7 +147,7 @@ controlnet_canny = ControlNetModel.from_pretrained(
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 
 pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-xl-base-1.0",
+    "black-forest-labs/FLUX.1-schnell",
     controlnet=controlnet,
     vae=vae,
     torch_dtype=torch.float16,
@@ -156,7 +156,7 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
 pipe.to(device)
 # Load model.
 pipe_canny = StableDiffusionXLControlNetPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-xl-base-1.0",
+    "black-forest-labs/FLUX.1-schnell",
     controlnet=controlnet_canny,
     vae=vae,
     safety_checker=None,