Upload 2 files

app.py

@@ -8,7 +8,7 @@ import onnxruntime as rt
 import pandas as pd
 from PIL import Image
 
-TITLE = "WaifuDiffusion Tagger"
+TITLE = "Tagger"
 DESCRIPTION = """
 """
 
@@ -66,9 +66,6 @@ def parse_args() -> argparse.Namespace:
 
 def load_labels(dataframe) -> list[str]:
     name_series = dataframe["name"]
-    #name_series = name_series.map(
-    #    lambda x: x.replace("_", " ") if x not in kaomojis else x
-    #)
     tag_names = name_series.tolist()
 
     rating_indexes = list(np.where(dataframe["category"] == 9)[0])
@@ -221,7 +218,11 @@ class Predictor:
             key=lambda x: x[1],
             reverse=True,
         )
+
         sorted_general_strings = [x[0] for x in sorted_general_strings]
+
+        sorted_general_strings = [x.replace("_", " ") if x not in kaomojis else x for x in sorted_general_strings]
+
         sorted_general_strings = (
             ", ".join(sorted_general_strings).replace("(", "\(").replace(")", "\)")
         )