check ignore

app.py

@@ -6,15 +6,20 @@ import gradio as gr
 import PIL.Image
 import zipfile
 from genTag import genTag
+from checkIgnore import is_ignore
 
 def predict(image: PIL.Image.Image, score_threshold: float):
     result_threshold = genTag(image, score_threshold)
     result_html = ''
     for label, prob in result_threshold.items():
-        result_html = result_html + '<p class="m5dd_list use"><span>' + str(label) + '</span><span>' + str(round(prob, 3)) + '</span></p>'
-    result_html = '<div>' + str(result_html) + '</div>'
-    result_text = ', '.join(result_threshold.keys())
-    result_text = '<div id="m5dd_result">' + str(result_text) + '</div>'
+        if is_ignore(label, 1):
+            result_html += '<p class="m5dd_list">'
+        else:
+            result_html += '<p class="m5dd_list use">'
+        result_html = result_html + '<span>' + str(label) + '</span><span>' + str(round(prob, 3)) + '</span></p>'
+    result_html = '<div>' + result_html + '</div>'
+    result_filter = {key: value for key, value in result_threshold.items() if not is_ignore(key, 1)}
+    result_text = '<div id="m5dd_result">' + ', '.join(result_filter.keys()) + '</div>'
     return result_html, result_text
 
 def predict_batch(zip_file, score_threshold: float, progress=gr.Progress()):
@@ -27,7 +32,8 @@ def predict_batch(zip_file, score_threshold: float, progress=gr.Progress()):
                 image = PIL.Image.open(image_file)
                 image = image.convert("RGB")
                 result_threshold = genTag(image, score_threshold)
-                tag = ', '.join(result_threshold.keys())
+                result_filter = {key: value for key, value in result_threshold.items() if not is_ignore(key, 2)}
+                tag = ', '.join(result_filter.keys())
                 result = result + str(file) + '\n' + str(tag) + '\n'
     return result
 
@@ -61,7 +67,9 @@ with gr.Blocks(css="style.css", js="script.js") as demo:
                                              value=0.5)
                 run_button2 = gr.Button('Run')
             with gr.Column(scale=2):
-                result_text2 = gr.Textbox(lines=5, show_copy_button=True)
+                result_text2 = gr.Textbox(lines=5,
+                                          label='Result',
+                                          show_copy_button=True)
 
     run_button.click(
         fn=predict,

checkIgnore.py

@@ -0,0 +1,22 @@
+#!/usr/bin/env python
+
+from __future__ import annotations
+
+def load_ignore(file_path):
+    result_dict = {}
+    with open(file_path, 'r') as file:
+        for line in file:
+            line = line.strip()
+            if line:
+                line_without_trailing_comma = line.rstrip(',')
+                result_dict[line_without_trailing_comma] = True
+    return result_dict
+
+ignore1 = load_ignore('ignoreTag.txt')
+ignore2 = load_ignore('ignoreTag2.txt')
+
+def is_ignore(tag, ignore_type):
+    if ignore_type == 1:
+        return tag in ignore1 or tag in ignore2
+    elif ignore_type == 2:
+        return tag in ignore1