Spaces:
Running
on
Zero
Running
on
Zero
tori29umai
commited on
Commit
•
d983f79
1
Parent(s):
fab4c46
Update utils/prompt_utils.py
Browse files- 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", "grey", "black"]
|
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)
|