narugo commited on
Commit
532d724
·
1 Parent(s): 0cc211e

dev(narugo): add konachan

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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