Update app.py

app.py

@@ -84,7 +84,7 @@ controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
     args.controlnet_model_name_or_path,
     revision=args.controlnet_revision,
     from_pt=args.controlnet_from_pt,
-    dtype=jnp.bfloat16,
+    dtype=jnp.float32,
 )
 
 pipeline, pipeline_params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
@@ -92,7 +92,7 @@ pipeline, pipeline_params = FlaxStableDiffusionControlNetPipeline.from_pretraine
     # tokenizer=tokenizer,
     controlnet=controlnet,
     safety_checker=None,
-    dtype=jnp.bfloat16,
+    dtype=jnp.float32,
     revision=args.revision,
     from_pt=args.from_pt,
 )
@@ -138,7 +138,6 @@ def infer(prompt, negative_prompt, image):
 
 with gr.Blocks(theme='gradio/soft') as demo:
     gr.Markdown("## Stable Diffusion with Hand Control")
-    gr.Markdown("In this app, you can find different ControlNets with different filters. ")
 
     with gr.Column():
         prompt_input = gr.Textbox(label="Prompt")