Change from canny to scribble

Changes the controlnet from canny to scribble.

app.py

@@ -14,7 +14,9 @@ import torch
 import requests
 from fastapi import FastAPI, HTTPException
 from PIL import Image
-from controlnet_aux import CannyDetector
+
+# from controlnet_aux import CannyDetector
+from controlnet_aux import ScribbleDetector
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 # model_repo_id = "stabilityai/sdxl-turbo"  # Replace to the model you would like to use
@@ -29,9 +31,9 @@ else:
 #     "lllyasviel/sd-controlnet-canny", torch_dtype=torch.float32
 # )
 
-controlnet = ControlNetModel.from_pretrained(
-    "diffusers/controlnet-canny-sdxl-1.0", torch_dtype=torch.float16
-)
+# controlnet = ControlNetModel.from_pretrained(
+#     "diffusers/controlnet-canny-sdxl-1.0", torch_dtype=torch.float16
+# )
 
 # pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 # pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
@@ -39,6 +41,20 @@ controlnet = ControlNetModel.from_pretrained(
 #     controlnet=controlnet,
 #     torch_dtype=torch_dtype,
 # ).to(device)
+# pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
+#     "stabilityai/stable-diffusion-xl-base-1.0",
+#     controlnet=controlnet,
+#     torch_dtype=torch.float16,
+#     variant="fp16",
+#     use_safetensors=True,
+# ).to(device)
+
+# Load SDXL-compatible ControlNet (scribble version)
+controlnet = ControlNetModel.from_pretrained(
+    "diffusers/controlnet-scribble-sdxl-1.0", torch_dtype=torch.float16
+)
+
+# Load SDXL base pipeline with the ControlNet
 pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
     controlnet=controlnet,
@@ -46,8 +62,9 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
     variant="fp16",
     use_safetensors=True,
 ).to(device)
-# pipe = pipe.to(device)
-canny = CannyDetector()
+
+# canny = CannyDetector()
+scribble_detector = ScribbleDetector()
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
@@ -96,7 +113,8 @@ def infer(
     # img = img.resize((req.width, req.height))
     img = img.resize((width, height))
 
-    control_net_image = canny(img).resize((width, height))
+    # control_net_image = canny(img).resize((width, height))
+    control_net_image = scribble_detector(img).resize((width, height))
 
     prompt = (
         "redraw the logo from scratch, clean sharp vector-style, "