k4d3
/

yiff_toolkit

Not-For-All-Audiences

Model card Files Files and versions Metrics Training metrics Community

k4d3 commited on Apr 21

Commit

448ca20

•

1 Parent(s): c513f21

awoo

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

Files changed (1) hide show

dataset_tools/Pony2Seaart.ipynb +1 -1

dataset_tools/Pony2Seaart.ipynb CHANGED Viewed

@@ -39,7 +39,7 @@
     "                else:\n",
     "                    tag = part.replace(\" \", \"_\")\n",
     "                    if tag == \"rating_explicit\":\n",
-    "                        tag = \"explicit\"\n",
     "                    elif tag == \"rating_safe\":\n",
     "                        tag = \"safe\"\n",
     "                    elif tag == \"rating_questionable\":\n",

     "                else:\n",
     "                    tag = part.replace(\" \", \"_\")\n",
     "                    if tag == \"rating_explicit\":\n",
+    "                        tag = \"nsfw\"\n",
     "                    elif tag == \"rating_safe\":\n",
     "                        tag = \"safe\"\n",
     "                    elif tag == \"rating_questionable\":\n",