Spaces:
Runtime error
Runtime error
LittleApple-fp16
commited on
Commit
•
29218b1
1
Parent(s):
0072751
Update waifu_get.py
Browse files- waifu_get.py +3 -2
waifu_get.py
CHANGED
@@ -8,6 +8,7 @@ from cyberharem.dataset.crawler import crawl_dataset_to_huggingface
|
|
8 |
import gradio as gr
|
9 |
import os
|
10 |
import json
|
|
|
11 |
from waifuc.action import HeadCountAction, AlignMinSizeAction, CCIPAction, ThreeStageSplitAction, ModeConvertAction, ClassFilterAction, PersonSplitAction, TaggingAction, RatingFilterAction, NoMonochromeAction, RandomFilenameAction, FirstNSelectAction, FilterSimilarAction, FileExtAction
|
12 |
from waifuc.export import SaveExporter, TextualInversionExporter
|
13 |
from waifuc.source import DanbooruSource, PixivSearchSource, ZerochanSource, LocalSource, GcharAutoSource
|
@@ -44,7 +45,7 @@ def start_func(chars, is_cpu, udghs, game_index=None):
|
|
44 |
# hf_hub_url(dgrepo, filename=os.path.relpath(name, dgrepo)),
|
45 |
repo_id=dgrepo, repo_type='dataset',
|
46 |
# os.path.join(os.path.basename(os.path.dirname(name)), 'pixiv_characters.json'),
|
47 |
-
filename=os.path.join(os.path.basename(os.path.dirname(name)), 'pixiv_characters.json'),
|
48 |
token=os.environ['HF_TOKEN']
|
49 |
)
|
50 |
# with open(os.path.join(os.path.basename(os.path.dirname(name)), 'pixiv_characters.json'), 'r') as f:
|
@@ -65,7 +66,7 @@ def start_func(chars, is_cpu, udghs, game_index=None):
|
|
65 |
# hf_hub_url(dgrepo, filename=os.path.relpath(name, dgrepo)),
|
66 |
repo_id=dgrepo, repo_type='dataset',
|
67 |
# os.path.join(os.path.basename(os.path.dirname(name)), 'pixiv_characters.json'),
|
68 |
-
filename=os.path.join(os.path.basename(os.path.dirname(name)), 'pixiv_characters.json'),
|
69 |
token=os.environ['HF_TOKEN']
|
70 |
)
|
71 |
# with open(os.path.join(os.path.basename(os.path.dirname(name)), 'pixiv_characters.json'), 'r') as f:
|
|
|
8 |
import gradio as gr
|
9 |
import os
|
10 |
import json
|
11 |
+
from pathlib import Path
|
12 |
from waifuc.action import HeadCountAction, AlignMinSizeAction, CCIPAction, ThreeStageSplitAction, ModeConvertAction, ClassFilterAction, PersonSplitAction, TaggingAction, RatingFilterAction, NoMonochromeAction, RandomFilenameAction, FirstNSelectAction, FilterSimilarAction, FileExtAction
|
13 |
from waifuc.export import SaveExporter, TextualInversionExporter
|
14 |
from waifuc.source import DanbooruSource, PixivSearchSource, ZerochanSource, LocalSource, GcharAutoSource
|
|
|
45 |
# hf_hub_url(dgrepo, filename=os.path.relpath(name, dgrepo)),
|
46 |
repo_id=dgrepo, repo_type='dataset',
|
47 |
# os.path.join(os.path.basename(os.path.dirname(name)), 'pixiv_characters.json'),
|
48 |
+
filename=Path(os.path.join(os.path.basename(os.path.dirname(name)), 'pixiv_characters.json')).as_posix(),
|
49 |
token=os.environ['HF_TOKEN']
|
50 |
)
|
51 |
# with open(os.path.join(os.path.basename(os.path.dirname(name)), 'pixiv_characters.json'), 'r') as f:
|
|
|
66 |
# hf_hub_url(dgrepo, filename=os.path.relpath(name, dgrepo)),
|
67 |
repo_id=dgrepo, repo_type='dataset',
|
68 |
# os.path.join(os.path.basename(os.path.dirname(name)), 'pixiv_characters.json'),
|
69 |
+
filename=Path(os.path.join(os.path.basename(os.path.dirname(name)), 'pixiv_characters.json')).as_posix(),
|
70 |
token=os.environ['HF_TOKEN']
|
71 |
)
|
72 |
# with open(os.path.join(os.path.basename(os.path.dirname(name)), 'pixiv_characters.json'), 'r') as f:
|