Update app.py

Trouble shooting

app.py

@@ -18,22 +18,22 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5", controlnet=controlnet, revision="flax", dtype=jnp.bfloat16
 )
 
-def infer(prompts, negative_prompts, image):
+def infer(prompt, image):
     params["controlnet"] = controlnet_params
     
     num_samples = 1 #jax.device_count()
     rng = create_key(0)
     rng = jax.random.split(rng, jax.device_count())
-    #im = canny_filter(image)
-    #canny_image = Image.fromarray(im)
+    im = image
+    image = Image.fromarray(im)
     
-    prompt_ids = pipe.prepare_text_inputs([prompts] * num_samples)
-    negative_prompt_ids = pipe.prepare_text_inputs([negative_prompts] * num_samples)
-    processed_image = pipe.prepare_image_inputs([canny_image] * num_samples)
+    #prompt_ids = pipe.prepare_text_inputs([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)
+    #prompt_ids = shard(prompt_ids)
+    #negative_prompt_ids = shard(negative_prompt_ids)
     processed_image = shard(processed_image)
     
     output = pipe(
@@ -42,7 +42,7 @@ def infer(prompts, negative_prompts, 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
     
@@ -52,7 +52,7 @@ def infer(prompts, negative_prompts, image):
 #gr.Interface(infer, inputs=["text", "text", "image"], outputs="gallery").launch()
 
 title = "Animal Pose Control Net"
-description = "This is a demo on ControlNet use animal keypoints."
+description = "This is a demo of Animal Pose ControlNet, which is a model trained on runwayml/stable-diffusion-v1-5 with new type of conditioning."
 
 with gr.Blocks(theme=gr.themes.Default(font=[gr.themes.GoogleFont("Inconsolata"), "Arial", "sans-serif"])) as demo:
     gr.Markdown(
@@ -68,7 +68,6 @@ gr.Examples(
             ["a tortoiseshell cat is sitting on a cushion",  "https://huggingface.co/JFoz/dog-cat-pose/blob/main/images_0.png"],
             ["a yellow dog standing on a lawn", "https://huggingface.co/JFoz/dog-cat-pose/blob/main/images_1.png"],
         ]
-        cache_examples=True,
     )
 
 gr.Interface(fn = infer, inputs = ["text", "text", "image"], outputs = "image",