Update app.py

app.py

@@ -18,7 +18,8 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5", controlnet=controlnet, revision="flax", dtype=jnp.bfloat16
 )
 
-def infer(prompt, image):
+#def infer(prompt, image):
+def infer(prompt):
     params["controlnet"] = controlnet_params
     
     num_samples = 1 #jax.device_count()