try to change t2i

app_haircolor_pix2pix.py

@@ -33,8 +33,10 @@ DEFAULT_NEGATIVE_PROMPT = "worst quality, normal quality, low quality, low res,
 
 DEFAULT_CATEGORY = "hair"
 
+canny_detector = CannyDetector()
+
 adapter = T2IAdapter.from_pretrained(
-    "TencentARC/t2iadapter_canny_sd15v2",
+    "TencentARC/t2i-adapter-canny-sdxl-1.0",
     torch_dtype=torch.float16,
     varient="fp16",
 )
@@ -67,7 +69,7 @@ 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 = custom_canny_detector(input_image)
+    canny_image = canny_detector(input_image)
 
     cond_image = canny_image
     cond_scale = cond_scale1
@@ -91,16 +93,6 @@ def image_to_image(
 
     return generated_image, time_cost_str
 
-def custom_canny_detector(image):
-    image = np.array(image)
-
-    low_threshold = 100
-    high_threshold = 200
-
-    image = cv2.Canny(image, low_threshold, high_threshold)
-    image = Image.fromarray(image)
-    return image
-
 def get_time_cost(run_task_time, time_cost_str):
     now_time = int(time.time()*1000)
     if run_task_time == 0: