Face-editor / prompts.py
Erwann Millon
initial commit
a23872f
raw
history blame
1.24 kB
import random
class PromptSet:
def __init__(self, pos, neg, config=None):
self.positive = pos
self.negative = neg
self.config = config
example_prompts = (
PromptSet("a picture of a woman with light blonde hair", "a picture of a person with dark hair | a picture of a person with brown hair"),
PromptSet("A picture of a woman with very thick eyebrows", "a picture of a person with very thin eyebrows | a picture of a person with no eyebrows"),
PromptSet("A picture of a woman wearing bright red lipstick", "a picture of a person wearing no lipstick | a picture of a person wearing dark lipstick"),
PromptSet("A picture of a beautiful chinese woman | a picture of a Japanese woman | a picture of an Asian woman", "a picture of a white woman | a picture of an Indian woman | a picture of a black woman"),
PromptSet("A picture of a handsome man | a picture of a masculine man", "a picture of a woman | a picture of a feminine person"),
PromptSet("A picture of a woman with a very big nose", "a picture of a person with a small nose | a picture of a person with a normal nose"),
)
def get_random_prompts():
prompt = random.choice(example_prompts)
return prompt.positive, prompt.negative