Spaces:
Running
Running
dev(narugo): add konachan
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ 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
|
@@ -31,6 +31,7 @@ _SITE_CLS = {
|
|
31 |
'zerochan': ZerochanWebpDataPool,
|
32 |
'gelbooru': GelbooruWebpDataPool,
|
33 |
'konachan': KonachanWebpDataPool,
|
|
|
34 |
}
|
35 |
|
36 |
|
|
|
9 |
import numpy as np
|
10 |
from PIL import Image
|
11 |
from cheesechaser.datapool import DanbooruWebpDataPool, YandeWebpDataPool, ZerochanWebpDataPool, GelbooruWebpDataPool, \
|
12 |
+
KonachanWebpDataPool, AnimePicturesWebpDataPool
|
13 |
from hfutils.operate import get_hf_fs, get_hf_client
|
14 |
from hfutils.utils import TemporaryDirectory
|
15 |
from imgutils.tagging import wd14
|
|
|
31 |
'zerochan': ZerochanWebpDataPool,
|
32 |
'gelbooru': GelbooruWebpDataPool,
|
33 |
'konachan': KonachanWebpDataPool,
|
34 |
+
'anime_pictures': AnimePicturesWebpDataPool,
|
35 |
}
|
36 |
|
37 |
|