awoo

Signed-off-by: Balazs Horvath <[email protected]>

README.md

@@ -263,8 +263,6 @@ This will put a link to the JSON next to the download button on e621.net and e6a
 
 ### [eva02-vit-large-448-8046](https://huggingface.co/Thouph/eva02-vit-large-448-8046)
 
-
-
 The following inference script for the tagger needs a folder as input, be warned that it also converts WebP images to PNG and you can specify tags to be ignored and some other stuff! I recommend reading through it and changing whatever you need.
 
 <div style="background-color: lightyellow; padding: 10px;">
@@ -392,6 +390,14 @@ for image_filename in image_files:
         and allowed_tags[indices[i]] not in ("placeholder0", "placeholder1")
     ]
 
+    # Replace 'safe', 'explicit', and 'questionable' with their 'rating_' counterparts
+    tags_to_write = [
+        tag.replace("safe", "rating_safe")
+        .replace("explicit", "rating_explicit")
+        .replace("questionable", "rating_questionable")
+        for tag in tags_to_write
+    ]
+
     # Escape unescaped parentheses in the tags
     tags_to_write_escaped = [
         re.sub(r"(?<!\\)(\()", r"\\\1", tag) for tag in tags_to_write