dev(narugo): add 2 more sites

app.py

@@ -8,7 +8,7 @@ import faiss
 import gradio as gr
 import numpy as np
 from PIL import Image
-from cheesechaser.datapool import DanbooruWebpDataPool, YandeWebpDataPool
+from cheesechaser.datapool import DanbooruWebpDataPool, YandeWebpDataPool, ZerochanWebpDataPool, GelbooruWebpDataPool
 from hfutils.operate import get_hf_fs, get_hf_client
 from hfutils.utils import TemporaryDirectory
 from imgutils.tagging import wd14
@@ -27,6 +27,8 @@ _ALL_MODEL_NAMES = [
 _SITE_CLS = {
     'danbooru': DanbooruWebpDataPool,
     'yandere': YandeWebpDataPool,
+    'zerochan': ZerochanWebpDataPool,
+    'gelbooru': GelbooruWebpDataPool,
 }