pngwn HF staff echarlaix HF staff commited on
Commit
47178c0
1 Parent(s): 4556988

modify-prompt (#4)

Browse files

- Modify prompt (c7a86e88a58b5f94991daf382b1540975759fbc7)
- Modify model (e547947a1b769b3af751bbabb2d13ab61fa235a0)
- Modify prompt (6ed68ef5d8239dc8c225958884f59f1165ab2d61)


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

Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -38,9 +38,8 @@ def fn(sign, cat):
38
 
39
  seed = random.randint(100, 1000000)
40
  set_seed(seed)
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
 
 
38
 
39
  seed = random.randint(100, 1000000)
40
  set_seed(seed)
41
+ response = pipe(starting_text + " " + sign + " art.", max_length=(len(starting_text) + random.randint(60, 90)), num_return_sequences=1)
42
+ pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
 
43
  image = pipe(response[0]["generated_text"], num_inference_steps=5).images[0]
44
  return [image, starting_text]
45