pngwn HF staff echarlaix HF staff commited on
Commit
4556988
1 Parent(s): 270595f

decrease-iteration (#3)

Browse files

- Decrease iteration number (952db281aeadce9722880f3e13c140085b65f729)


Co-authored-by: Ella Charlaix <echarlaix@users.noreply.huggingface.co>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def fn(sign, cat):
41
 
42
  response = pipe(starting_text, max_length=(len(starting_text) + random.randint(60, 90)), num_return_sequences=1)
43
  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
44
- image = pipe(response[0]["generated_text"], num_inference_steps=10).images[0]
45
  return [image, starting_text]
46
 
47
 
 
41
 
42
  response = pipe(starting_text, max_length=(len(starting_text) + random.randint(60, 90)), num_return_sequences=1)
43
  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
44
+ image = pipe(response[0]["generated_text"], num_inference_steps=5).images[0]
45
  return [image, starting_text]
46
 
47