Update app.py

app.py

@@ -20,7 +20,7 @@ pipe, params = FlaxStableDiffusionPipeline.from_pretrained(
 def infer(prompts, negative_prompts, width=1088, height=1088, inference_steps=30, seed=0):
     
     num_samples = 1 #jax.device_count()
-    rng = create_key(seed)
+    rng = create_key(int(seed))
     rng = jax.random.split(rng, jax.device_count())
     
     prompt_ids = pipe.prepare_inputs([prompts] * num_samples)