Update app.py

app.py

@@ -20,14 +20,13 @@ from typing import Optional, List
 from tqdm import tqdm
 import supervision as sv
 
-
 accelerator = Accelerator(cpu=True)
+controlnet = accelerator.prepare(ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.bfloat16))
 pipe = accelerator.prepare(StableDiffusionControlNetImg2ImgPipeline.from_pretrained("stabilityai/sdxl-turbo", controlnet=controlnet, torch_dtype=torch.bfloat16, use_safetensors=False, variant="fp16", safety_checker=None))
 pipe.unet.to(memory_format=torch.channels_last)
 pipe.scheduler = accelerator.prepare(EulerDiscreteScheduler.from_config(pipe.scheduler.config))
 pipe = pipe.to("cpu")
 
-controlnet = accelerator.prepare(ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.bfloat16))
 def plex(fpath, text, neg_prompt, one, two, three, four, five):
     gc.collect()
     prompt = text