diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-12-24 12:40:32 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-12-24 16:22:47 +0300 |
| commit | 03d7b394539558f6f560155d87a4fc66eb675e30 (patch) | |
| tree | 26d27dec0b8f4b4b0273f0a173372d71c518e2e6 /modules/deepbooru.py | |
| parent | ccaacb1891f018d380880ea4deb4ab79ad33d0d9 (diff) | |
added an option to filter out deepbooru tags
Diffstat (limited to 'modules/deepbooru.py')
| -rw-r--r-- | modules/deepbooru.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/deepbooru.py b/modules/deepbooru.py index dfc83357..122fce7f 100644 --- a/modules/deepbooru.py +++ b/modules/deepbooru.py @@ -79,7 +79,9 @@ class DeepDanbooru: res = [] - for tag in tags: + filtertags = set([x.strip().replace(' ', '_') for x in shared.opts.deepbooru_filter_tags.split(",")]) + + for tag in [x for x in tags if x not in filtertags]: probability = probability_dict[tag] tag_outformat = tag if use_spaces: |