Spaces:
Sleeping
Sleeping
decrease-iteration (#3)
Browse files- Decrease iteration number (952db281aeadce9722880f3e13c140085b65f729)
Co-authored-by: Ella Charlaix <echarlaix@users.noreply.huggingface.co>
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=
|
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 |
|