Spaces:
Running
Running
dev(narugo): use dgzyka as default
Browse files
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
|
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 = '
|
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')
|