Spaces:
Runtime error
Runtime error
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:
|