aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGreendayle <Greendayle>2022-10-05 21:15:08 +0200
committerGreendayle <Greendayle>2022-10-05 21:15:08 +0200
commit1506fab29ad54beb9f52236912abc432209c8089 (patch)
tree0f8d0c66955345c0a63823ef0fc91ae5be83aa70
parent59a2b9e5afc27d2fda72069ca0635070535d18fe (diff)
removing problematic tag
-rw-r--r--modules/deepbooru.py5
1 files changed, 2 insertions, 3 deletions
diff --git a/modules/deepbooru.py b/modules/deepbooru.py
index 958b1c3d..841cb9c5 100644
--- a/modules/deepbooru.py
+++ b/modules/deepbooru.py
@@ -38,13 +38,12 @@ def _load_tf_and_return_tags(pil_image, threshold):
for i, tag in enumerate(tags):
result_dict[tag] = y[i]
-
-
-
result_tags_out = []
result_tags_print = []
for tag in tags:
if result_dict[tag] >= threshold:
+ if tag.startswith("rating:"):
+ continue
result_tags_out.append(tag)
result_tags_print.append(f'{result_dict[tag]} {tag}')