Update app.py

Adding negative prompts back

app.py

@@ -18,23 +18,22 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5", controlnet=controlnet, revision="flax", dtype=jnp.bfloat16
 )
 
-def infer(prompt, image):
-#def infer(prompt):
+def infer(prompts, negative_prompts, image):
+
     params["controlnet"] = controlnet_params
     
     num_samples = 1 #jax.device_count()
     rng = create_key(0)
     rng = jax.random.split(rng, jax.device_count())
-    im = image
-    image = Image.fromarray(im)
+    image = Image.fromarray(image)
     
     prompt_ids = pipe.prepare_text_inputs([prompts] * num_samples)
-    #negative_prompt_ids = pipe.prepare_text_inputs([negative_prompts] * num_samples)
+    negative_prompt_ids = pipe.prepare_text_inputs([negative_prompts] * num_samples)
     processed_image = pipe.prepare_image_inputs([image] * num_samples)
     
     p_params = replicate(params)
     prompt_ids = shard(prompt_ids)
-    #negative_prompt_ids = shard(negative_prompt_ids)
+    negative_prompt_ids = shard(negative_prompt_ids)
     processed_image = shard(processed_image)
     
     output = pipe(
@@ -43,7 +42,7 @@ def infer(prompt, image):
         params=p_params,
         prng_seed=rng,
         num_inference_steps=50,
-        #neg_prompt_ids=negative_prompt_ids,
+        neg_prompt_ids=negative_prompt_ids,
         jit=True,
     ).images