search_image_by_image

Running

narugo commited on 12 days ago

Commit

71df424

•

1 Parent(s): a7dab88

dev(narugo): add newest

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 from cheesechaser.datapool import DanbooruWebpDataPool, YandeWebpDataPool, ZerochanWebpDataPool, GelbooruWebpDataPool, \
-    KonachanWebpDataPool, AnimePicturesWebpDataPool
 from hfutils.operate import get_hf_fs, get_hf_client
 from hfutils.utils import TemporaryDirectory
 from imgutils.tagging import wd14
@@ -26,7 +26,7 @@ _ALL_MODEL_NAMES = [
 ]
 _SITE_CLS = {
-    'danbooru': DanbooruWebpDataPool,
     'yandere': YandeWebpDataPool,
     'zerochan': ZerochanWebpDataPool,
     'gelbooru': GelbooruWebpDataPool,

 import numpy as np
 from PIL import Image
 from cheesechaser.datapool import DanbooruWebpDataPool, YandeWebpDataPool, ZerochanWebpDataPool, GelbooruWebpDataPool, \
+    KonachanWebpDataPool, AnimePicturesWebpDataPool, DanbooruNewestWebpDataPool
 from hfutils.operate import get_hf_fs, get_hf_client
 from hfutils.utils import TemporaryDirectory
 from imgutils.tagging import wd14
 ]
 _SITE_CLS = {
+    'danbooru': DanbooruNewestWebpDataPool,
     'yandere': YandeWebpDataPool,
     'zerochan': ZerochanWebpDataPool,
     'gelbooru': GelbooruWebpDataPool,