Upload 3 files

app.py

@@ -11,6 +11,8 @@ from PIL import Image
 TITLE = "Tagger"
 DESCRIPTION = """
 """
+implications_list_path = './implications_list.json'
+related_feature_path = './related_feature.json'
 
 #HF_TOKEN = os.environ["HF_TOKEN"]
 
@@ -60,6 +62,7 @@ def parse_args() -> argparse.Namespace:
     parser.add_argument("--score-slider-step", type=float, default=0.05)
     parser.add_argument("--score-general-threshold", type=float, default=0.4)
     parser.add_argument("--score-character-threshold", type=float, default=0.9)
+    parser.add_argument("--character_string", type=str)
     parser.add_argument("--share", action="store_true")
     return parser.parse_args()
 
@@ -166,6 +169,7 @@ class Predictor:
         general_mcut_enabled,
         character_thresh,
         character_mcut_enabled,
+        character_string,
     ):
         self.load_model(model_repo)
 
@@ -191,7 +195,11 @@ class Predictor:
         general_res = [x for x in general_names if x[1] > general_thresh]
         general_res = dict(general_res)
 
-        with open('./implications_list.json', 'r') as f:
+        with open(related_feature_path, 'r') as f:
+            related_feature_list = json.load(f)
+
+
+        with open(implications_list_path, 'r') as f:
             implications_list = json.load(f)
 
         to_delete = set()
@@ -219,8 +227,17 @@ class Predictor:
             reverse=True,
         )
 
+        character_list = character_string.lower().split(', ')
+
+        feature_delete_list = []
+        for tag in character_list:
+            if tag in related_feature_list:  
+                feature_delete_list.extend(related_feature_list[tag])
+
         sorted_general_strings = [x[0] for x in sorted_general_strings]
 
+        sorted_general_strings = [x for x in sorted_general_strings if x not in feature_delete_list]
+
         sorted_general_strings = [x.replace("_", " ") if x not in kaomojis else x for x in sorted_general_strings]
 
         sorted_general_strings = (
@@ -284,15 +301,16 @@ def main():
                             label="Use MCut threshold",
                             scale=1,
                         )
+                    with gr.Row():
+                        character_string = gr.Textbox(
+                        label= "Character",
+                        scale=3,
+                        )
                     with gr.Row():
                         clear = gr.ClearButton(
                             components=[
                                 image,
-                                model_repo,
-                                general_thresh,
-                                general_mcut_enabled,
-                                character_thresh,
-                                character_mcut_enabled,
+                                character_string,
                             ],
                             variant="secondary",
                             size="lg",
@@ -321,6 +339,7 @@ def main():
                 general_mcut_enabled,
                 character_thresh,
                 character_mcut_enabled,
+                character_string,
             ],
             outputs=[sorted_general_strings, rating, character_res, general_res],
         )