bguisard commited on
Commit
c4c4b83
1 Parent(s): c8a989c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ pipeline, pipeline_params = FlaxStableDiffusionPipeline.from_pretrained(
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)
14
  images = pipeline(
@@ -43,7 +43,7 @@ app = gr.Interface(
43
  "Stable Diffusion Nano allows for fast prototyping of diffusion models, "
44
  "enabling quick experimentation with easily available hardware."
45
  ),
46
- examples=[["A watercolor painting of a bird"]],
47
  )
48
 
49
  app.launch()
 
8
 
9
 
10
  def generate_image(prompt: str, inference_steps: int = 30, prng_seed: int = 0):
11
+ rng = jax.random.PRNGKey(int(prng_seed))
12
 
13
  prompt_ids = pipeline.prepare_inputs(prompt)
14
  images = pipeline(
 
43
  "Stable Diffusion Nano allows for fast prototyping of diffusion models, "
44
  "enabling quick experimentation with easily available hardware."
45
  ),
46
+ examples=[["A watercolor painting of a bird", 30, 0]],
47
  )
48
 
49
  app.launch()