Update app.py

app.py

@@ -50,7 +50,8 @@ def infer(prompts, negative_prompts, image):
     num_samples = jax.device_count()
     rng = create_key(0)
     rng = jax.random.split(rng, jax.device_count())
-    canny_image = canny_filter(image)
+    im = canny_filter(image)
+    canny_image = Image.fromarray(im)
     
     prompt_ids = pipe.prepare_text_inputs([prompts] * num_samples)
     negative_prompt_ids = pipe.prepare_text_inputs([negative_prompts] * num_samples)