alfredplpl commited on
Commit
42a9377
1 Parent(s): 361d7a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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=["boy" if p=="1boy" else p for p in splited_prompt]
82
- human_words=["girl","maid","female","woman","boy","male","man","guy"]
 
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"