Commit
·
0cc211e
1
Parent(s):
36f33a6
dev(narugo): add konachan
Browse files
app.py
CHANGED
|
@@ -8,7 +8,8 @@ import faiss
|
|
| 8 |
import gradio as gr
|
| 9 |
import numpy as np
|
| 10 |
from PIL import Image
|
| 11 |
-
from cheesechaser.datapool import DanbooruWebpDataPool, YandeWebpDataPool, ZerochanWebpDataPool, GelbooruWebpDataPool
|
|
|
|
| 12 |
from hfutils.operate import get_hf_fs, get_hf_client
|
| 13 |
from hfutils.utils import TemporaryDirectory
|
| 14 |
from imgutils.tagging import wd14
|
|
@@ -29,6 +30,7 @@ _SITE_CLS = {
|
|
| 29 |
'yandere': YandeWebpDataPool,
|
| 30 |
'zerochan': ZerochanWebpDataPool,
|
| 31 |
'gelbooru': GelbooruWebpDataPool,
|
|
|
|
| 32 |
}
|
| 33 |
|
| 34 |
|
|
|
|
| 8 |
import gradio as gr
|
| 9 |
import numpy as np
|
| 10 |
from PIL import Image
|
| 11 |
+
from cheesechaser.datapool import DanbooruWebpDataPool, YandeWebpDataPool, ZerochanWebpDataPool, GelbooruWebpDataPool, \
|
| 12 |
+
KonachanWebpDataPool
|
| 13 |
from hfutils.operate import get_hf_fs, get_hf_client
|
| 14 |
from hfutils.utils import TemporaryDirectory
|
| 15 |
from imgutils.tagging import wd14
|
|
|
|
| 30 |
'yandere': YandeWebpDataPool,
|
| 31 |
'zerochan': ZerochanWebpDataPool,
|
| 32 |
'gelbooru': GelbooruWebpDataPool,
|
| 33 |
+
'konachan': KonachanWebpDataPool,
|
| 34 |
}
|
| 35 |
|
| 36 |
|