Update app.py

app.py

@@ -12,7 +12,7 @@ pipeline, pipeline_params = FlaxStableDiffusionPipeline.from_pretrained(
 def generate_image(prompt: str, inference_steps: int = 30, prng_seed: int = 0):
     rng = jax.random.PRNGKey(int(prng_seed))
     rng = jax.random.split(rng, jax.device_count())
-    p_params = replicate(params)
+    p_params = replicate(pipeline_params)
     
     num_samples = 1
     prompt_ids = pipeline.prepare_inputs([prompt] * num_samples)