Update app.py

app.py

@@ -7,7 +7,7 @@ from flax.training.common_utils import shard
 from diffusers.utils import load_image
 from PIL import Image
 from diffusers import FlaxStableDiffusionControlNetPipeline, FlaxControlNetModel
-
+import cv2
 
 def image_grid(imgs, rows, cols):
     w, h = imgs[0].size
@@ -24,10 +24,14 @@ def create_key(seed=0):
 rng = create_key(0)
 
 
-def canny_filter(image):
 
-    ## TODO: Implement canny filter here
-    return canny_image
+
+def canny_filter(image):
+    gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    
+    blurred_image = cv2.GaussianBlur(gray_image, (5, 5), 0)
+    edges_image = cv2.Canny(blurred_image, 50, 150)
+    return edges_image
 
 def infer(prompts, negative_prompts, image):