Update README.md

README.md

@@ -1,8 +1,17 @@
 ---
 license: cc0-1.0
 widget:
-- src: https://huggingface.co/carbon225/vit-base-patch16-224-hentai/resolve/main/samples/1.jpeg
-- src: https://huggingface.co/carbon225/vit-base-patch16-224-hentai/resolve/main/samples/2.jpeg
+- src: >-
+    https://huggingface.co/carbon225/vit-base-patch16-224-hentai/resolve/main/samples/1.jpeg
+- src: >-
+    https://huggingface.co/carbon225/vit-base-patch16-224-hentai/resolve/main/samples/2.jpeg
+datasets:
+- carbon225/vndb_img
+tags:
+- art
+- anime
+- visual-novel
+- nsfw
 ---
 
 # ViT for NSFW classification
@@ -17,8 +26,8 @@ finetuned for flagging images according to [vndb.org](https://vndb.org/d19) with
 ## Training data
 The model was trained on the vndb.org [database dump](https://vndb.org/d14)
 using full size screenshots (`sf` in the database dump).
-Because the dataset contains questionable images, I will not publish it.
+The dataset can be loaded from [carbon225/vndb_img](https://huggingface.co/datasets/carbon225/vndb_img).
 
 ## Intended use
 The model can be used for flagging anime-style images for sexual content.
-It can also be finetuned on other tasks related to anime images.
+It can also be finetuned on other tasks related to anime images.