narugo commited on
Commit
a54d829
1 Parent(s): 71df424

dev(narugo): use dgzyka as default

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ 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
  KonachanWebpDataPool, AnimePicturesWebpDataPool, DanbooruNewestWebpDataPool
13
  from hfutils.operate import get_hf_fs, get_hf_client
14
  from hfutils.utils import TemporaryDirectory
@@ -19,7 +19,7 @@ _REPO_ID = 'deepghs/index_experiments'
19
  hf_fs = get_hf_fs()
20
  hf_client = get_hf_client()
21
 
22
- _DEFAULT_MODEL_NAME = 'SwinV2_v3_gzyka_15320102_4GB'
23
  _ALL_MODEL_NAMES = [
24
  os.path.dirname(os.path.relpath(path, _REPO_ID))
25
  for path in hf_fs.glob(f'{_REPO_ID}/*/knn.index')
 
8
  import gradio as gr
9
  import numpy as np
10
  from PIL import Image
11
+ from cheesechaser.datapool import YandeWebpDataPool, ZerochanWebpDataPool, GelbooruWebpDataPool, \
12
  KonachanWebpDataPool, AnimePicturesWebpDataPool, DanbooruNewestWebpDataPool
13
  from hfutils.operate import get_hf_fs, get_hf_client
14
  from hfutils.utils import TemporaryDirectory
 
19
  hf_fs = get_hf_fs()
20
  hf_client = get_hf_client()
21
 
22
+ _DEFAULT_MODEL_NAME = 'SwinV2_v3_dgzyka_23325111_8GB'
23
  _ALL_MODEL_NAMES = [
24
  os.path.dirname(os.path.relpath(path, _REPO_ID))
25
  for path in hf_fs.glob(f'{_REPO_ID}/*/knn.index')