merve HF staff commited on
Commit
84ad302
1 Parent(s): e281593

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -50,7 +50,8 @@ def infer(prompts, negative_prompts, image):
50
  num_samples = jax.device_count()
51
  rng = create_key(0)
52
  rng = jax.random.split(rng, jax.device_count())
53
- canny_image = canny_filter(image)
 
54
 
55
  prompt_ids = pipe.prepare_text_inputs([prompts] * num_samples)
56
  negative_prompt_ids = pipe.prepare_text_inputs([negative_prompts] * num_samples)
 
50
  num_samples = jax.device_count()
51
  rng = create_key(0)
52
  rng = jax.random.split(rng, jax.device_count())
53
+ im = canny_filter(image)
54
+ canny_image = Image.fromarray(im)
55
 
56
  prompt_ids = pipe.prepare_text_inputs([prompts] * num_samples)
57
  negative_prompt_ids = pipe.prepare_text_inputs([negative_prompts] * num_samples)