Update app.py

app.py

@@ -6,13 +6,17 @@ from pytorch_lightning import seed_everything
 from annotator.util import resize_image, HWC3
 from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
 
-# Load the controlnet model
-controlnet = ControlNetModel.from_pretrained("CompVis/controlnet")
+# # Load the controlnet model
+# controlnet = ControlNetModel.from_pretrained("CompVis/controlnet")
 
-# Load the pipeline
+# # Load the pipeline
+# pipe = StableDiffusionControlNetPipeline.from_pretrained(
+#     "CompVis/stable-diffusion-v1-4",
+#     controlnet=controlnet
+# ).to("cuda")
+controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.float16)
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
-    "CompVis/stable-diffusion-v1-4",
-    controlnet=controlnet
+    "runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
 ).to("cuda")
 
 def process(input_image, prompt, a_prompt, n_prompt, num_samples, image_resolution, ddim_steps, guess_mode, strength, scale, seed, eta, low_threshold, high_threshold):