LittleApple-fp16 commited on
Commit
29218b1
1 Parent(s): 0072751

Update waifu_get.py

Browse files
Files changed (1) hide show
  1. 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: