Commit 1506fab2 authored by Greendayle's avatar Greendayle

removing problematic tag

parent 59a2b9e5
...@@ -38,13 +38,12 @@ def _load_tf_and_return_tags(pil_image, threshold): ...@@ -38,13 +38,12 @@ def _load_tf_and_return_tags(pil_image, threshold):
for i, tag in enumerate(tags): for i, tag in enumerate(tags):
result_dict[tag] = y[i] result_dict[tag] = y[i]
result_tags_out = [] result_tags_out = []
result_tags_print = [] result_tags_print = []
for tag in tags: for tag in tags:
if result_dict[tag] >= threshold: if result_dict[tag] >= threshold:
if tag.startswith("rating:"):
continue
result_tags_out.append(tag) result_tags_out.append(tag)
result_tags_print.append(f'{result_dict[tag]} {tag}') result_tags_print.append(f'{result_dict[tag]} {tag}')
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment