echarlaix HF staff commited on
Commit
6ed68ef
1 Parent(s): e547947

Modify prompt

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -38,11 +38,9 @@ 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("runwayml/stable-diffusion-v1-5")
44
- # image = pipe(response[0]["generated_text"], num_inference_steps=5).images[0]
45
- image = pipe(sign + " art", num_inference_steps=5).images[0]
46
  return [image, starting_text]
47
 
48
 
 
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
 
46