dev(narugo): add rule34

app.py

@@ -9,7 +9,7 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 from cheesechaser.datapool import YandeWebpDataPool, ZerochanWebpDataPool, GelbooruWebpDataPool, \
-    KonachanWebpDataPool, AnimePicturesWebpDataPool, DanbooruNewestWebpDataPool
+    KonachanWebpDataPool, AnimePicturesWebpDataPool, DanbooruNewestWebpDataPool, Rule34WebpDataPool
 from hfutils.operate import get_hf_fs, get_hf_client
 from hfutils.utils import TemporaryDirectory
 from imgutils.tagging import wd14
@@ -32,6 +32,7 @@ _SITE_CLS = {
     'gelbooru': GelbooruWebpDataPool,
     'konachan': KonachanWebpDataPool,
     'anime_pictures': AnimePicturesWebpDataPool,
+    'rule34': Rule34WebpDataPool,
 }
 
 

requirements.txt

@@ -3,4 +3,4 @@ faiss-cpu
 dghs-imgutils
 onnxruntime
 gradio==5.5.0
-cheesechaser
+cheesechaser>=0.1.6