tori29umai commited on
Commit
9680e12
1 Parent(s): 2919701

Update utils/prompt_utils.py

Browse files
Files changed (1) hide show
  1. utils/prompt_utils.py +1 -1
utils/prompt_utils.py CHANGED
@@ -14,7 +14,7 @@ def remove_duplicates(base_prompt):
14
  def remove_color(base_prompt):
15
  # タグの色情報を取り除く
16
  prompt_list = base_prompt.split(", ")
17
- color_list = ["pink", "red", "orange", "brown", "yellow", "green", "blue", "purple", "blonde", "colored skin", "white hair"]
18
  # カラータグを除去します。
19
  cleaned_tags = [tag for tag in prompt_list if all(color.lower() not in tag.lower() for color in color_list)]
20
  return ", ".join(cleaned_tags)
 
14
  def remove_color(base_prompt):
15
  # タグの色情報を取り除く
16
  prompt_list = base_prompt.split(", ")
17
+ color_list = ["pink", "red", "orange", "brown", "yellow", "green", "blue", "purple", "blonde", "colored skin", "white hair", "gray"]
18
  # カラータグを除去します。
19
  cleaned_tags = [tag for tag in prompt_list if all(color.lower() not in tag.lower() for color in color_list)]
20
  return ", ".join(cleaned_tags)