Spaces:
Running
on
Zero
Running
on
Zero
alfredplpl
commited on
Commit
·
42a9377
1
Parent(s):
361d7a3
Update app.py
Browse files
app.py
CHANGED
@@ -78,8 +78,9 @@ def auto_prompt_correction(prompt_ui,neg_prompt_ui,cool_japan_type_ui):
|
|
78 |
splited_prompt=prompt.replace(","," ").replace("_"," ").split(" ")
|
79 |
splited_prompt=["a person" if p=="solo" else p for p in splited_prompt]
|
80 |
splited_prompt=["girl" if p=="1girl" else p for p in splited_prompt]
|
81 |
-
splited_prompt=["
|
82 |
-
|
|
|
83 |
for word in human_words:
|
84 |
if( word in splited_prompt):
|
85 |
prompt=f"{cool_japan_type}, {prompt}, 4k, detailed"
|
|
|
78 |
splited_prompt=prompt.replace(","," ").replace("_"," ").split(" ")
|
79 |
splited_prompt=["a person" if p=="solo" else p for p in splited_prompt]
|
80 |
splited_prompt=["girl" if p=="1girl" else p for p in splited_prompt]
|
81 |
+
splited_prompt=["a couple of girls" if p=="2girls" else p for p in splited_prompt]
|
82 |
+
splited_prompt=["a couple of boys" if p=="2boys" else p for p in splited_prompt]
|
83 |
+
human_words=["girl","maid","maids","female","woman","girls","women","boy","boys","male","man","men","guy","guys"]
|
84 |
for word in human_words:
|
85 |
if( word in splited_prompt):
|
86 |
prompt=f"{cool_japan_type}, {prompt}, 4k, detailed"
|