narugo commited on
Commit
0cc211e
·
1 Parent(s): 36f33a6

dev(narugo): add konachan

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