Update app.py

app.py

@@ -14,7 +14,7 @@ import random
 
 os.system("pip install -e ./controlnet_aux")
 
-from controlnet_aux import OpenposeDetector, CannyDetector
+from controlnet_aux import OpenposeDetector #, CannyDetector
 from depth_anything_v2.dpt import DepthAnythingV2
 
 from huggingface_hub import hf_hub_download
@@ -113,7 +113,7 @@ strength_mapping = {
     "pose": 1.0,
 }
 
-canny = CannyDetector()
+# canny = CannyDetector()
 open_pose = OpenposeDetector.from_pretrained("lllyasviel/Annotators")
 
 torch.backends.cuda.matmul.allow_tf32 = True
@@ -139,9 +139,13 @@ def extract_openpose(img):
     processed_image_open_pose = open_pose(img, hand_and_face=True)
     return processed_image_open_pose
     
-def extract_canny(image):
-    processed_image_canny = canny(image)
-    return processed_image_canny
+def extract_canny(input_image):
+    image = np.array(input_image)
+    image = cv2.Canny(image, self.low_threshold, self.high_threshold)
+    image = image[:, :, None]
+    image = np.concatenate([image, image, image], axis=2)
+    canny_image = Image.fromarray(image)
+    return canny_image
 
 def apply_gaussian_blur(image, kernel_size=(21, 21)):
     image = convert_from_image_to_cv2(image)