Update app.py

change dtype back

app.py

@@ -29,10 +29,10 @@ sketch_url = 'https://editor.p5js.org/kfahn/full/Ntzq9HWhx'
 control_img = 'myimage.jpg'
 
 controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
-    "JFoz/dog-cat-pose", dtype=jnp.bfloat32
+    "JFoz/dog-cat-pose", dtype=jnp.bfloat16
 )
 pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5", controlnet=controlnet, revision="flax", dtype=jnp.bfloat32
+    "runwayml/stable-diffusion-v1-5", controlnet=controlnet, revision="flax", dtype=jnp.bfloat16
 )
 
 def infer(prompts, negative_prompts, image):