Update app.py

app.py

@@ -75,17 +75,17 @@ def pipe_inference(
     if not isinstance(image, np.ndarray):
         image = np.array(image) 
 
-    resized_image = resize_image(image, resolution)
+    processed_image = resize_image(image, resolution)
         
     if not is_canny:
-        resized_image = preprocess_canny(resized_image, resolution)
+        resized_image, processed_image = preprocess_canny(processed_image, resolution)
 
     rng = create_key(seed)
     rng = jax.random.split(rng, jax.device_count())
 
     prompt_ids = pipe.prepare_text_inputs([prompt] * num_samples)
     negative_prompt_ids = pipe.prepare_text_inputs([negative_prompt] * num_samples)
-    processed_image = pipe.prepare_image_inputs([resized_image] * num_samples)
+    processed_image = pipe.prepare_image_inputs([processed_image] * num_samples)
         
     p_params = replicate(params)
     prompt_ids = shard(prompt_ids)