Naturalize composition frame wording

This commit is contained in:
2026-06-27 21:45:21 +02:00
parent e9bd9c45ca
commit a4b4dae8cf
7 changed files with 50 additions and 7 deletions
+4 -1
View File
@@ -145,7 +145,10 @@ def metadata_action_label(row: dict[str, Any], default: str = "sexual pose") ->
def normalize_composition(text: str) -> str:
return re.sub(r"^vertical\s+", "", input_policy.clean_text(text), flags=re.IGNORECASE)
text = re.sub(r"^vertical\s+", "", input_policy.clean_text(text), flags=re.IGNORECASE)
text = re.sub(r"\s+composition$", "", text, flags=re.IGNORECASE)
text = re.sub(r"\bcomposition\b", "frame", text, flags=re.IGNORECASE)
return text.strip(" ,")
def clean_clothing(text: str) -> str: