change adapter

app_haircolor_inpaint_adapter_15.py

@@ -58,7 +58,7 @@ pidiNet_detector = pidiNet_detector.to(DEVICE)
 # )
 # adapters = adapters.to(torch.float16)
 adapters = T2IAdapter.from_pretrained(
-    "TencentARC/t2iadapter_canny_sd15v2",
+    "TencentARC/t2iadapter_sketch_sd15v2",
     torch_dtype=torch.float16,
     varient="fp16",
 )
@@ -92,8 +92,8 @@ def image_to_image(
     run_task_time = 0
     time_cost_str = ''
     run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
-    canny_image = canny_detector(input_image, int(generate_size*1), generate_size)
-    canny_image = canny_image.convert("L")
+    # canny_image = canny_detector(input_image, int(generate_size*1), generate_size)
+    canny_image = custom_canny_detector(input_image)
     # lineart_image = lineart_detector(input_image, int(generate_size*1), generate_size)
     # run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
     pidiNet_image = pidiNet_detector(input_image, int(generate_size*1), generate_size)