Update app.py

app.py

@@ -61,4 +61,7 @@ pipeline, params = FlaxStableDiffusionPipeline.from_pretrained(
     use_auth_token = HF_ACCESS_TOKEN,
     revision="bf16",
     dtype=jnp.bfloat16,
-)
+)
+
+prompts = ["apple"] * 1
+images = sd2_inference(pipeline, prompts, params, seed = 42, num_inference_steps = 1 )