Spaces:
Sleeping
Sleeping
Modify prompt
Browse files
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 |
-
|
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 |
|