Add skip and translate functionality to predict function

app.py

@@ -46,6 +46,16 @@ def load_labels() -> list[str]:
 model = load_model()
 labels = load_labels()
 
+skip = ["rating:safe",
+        "rating:questionable",
+        "rating:explicit",
+        "3d", 
+        "photorealistic", 
+        "realistic",
+        "uncensored"]
+
+translate = {'yuri': 'lesbian'}
+
 
 def predict(
         image: PIL.Image.Image, score_threshold: float
@@ -68,6 +78,11 @@ def predict(
     for index in indices:
         label = labels[index]
         prob = probs[index]
+        if label in skip:
+            continue
+        if label in translate:
+            label = translate[label]
+
         result_all[label] = prob
         if prob < score_threshold:
             break