JairoDanielMT's picture
Upload folder using huggingface_hub
4ef6c2b verified
Raw
History Blame Contribute Delete
344 Bytes
class Normalizer:
def __init__(self):
self.rules = {
"orange sweater": "orange_turtleneck",
"orange pullover": "orange_turtleneck",
"pink hair": "pink_quills"
}
def normalize(self, tag: str) -> str:
return self.rules.get(tag.lower(), tag.lower().replace(" ", "_"))