bguisard commited on
Commit
bb347bc
1 Parent(s): 894da8d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ pipeline, pipeline_params = FlaxStableDiffusionPipeline.from_pretrained(
7
  )
8
 
9
 
10
- def generate_image(prompt: str, inference_steps: int, prng_seed: int = 0):
11
  rng = jax.random.PRNGKey(prng_seed)
12
 
13
  prompt_ids = pipeline.prepare_inputs(prompt)
 
7
  )
8
 
9
 
10
+ def generate_image(prompt: str, inference_steps: int = 30, prng_seed: int = 0):
11
  rng = jax.random.PRNGKey(prng_seed)
12
 
13
  prompt_ids = pipeline.prepare_inputs(prompt)