memoize and also based

app.py

@@ -7,7 +7,20 @@ from PIL import Image
 from huggingface_hub import hf_hub_download
 from onnxruntime import InferenceSession
 
-
+from functools import wraps
+
+def memoize(func):
+    cache = {}
+    cache_maxsize = 25
+    @wraps(func)
+    def memoized_func(*args):
+        if args not in cache:
+            cache[args] = func(*args)
+            if len(cache) > cache_maxsize:
+                cache.popitem(last=False)
+        return cache[args]
+
+    return memoized_func
 # noinspection PyUnresolvedReferences
 def make_square(img, target_size):
     old_size = img.shape[:2]
@@ -119,7 +132,7 @@ WAIFU_MODELS: Mapping[str, WaifuDiffusionInterrogator] = {
 }
 RE_SPECIAL = re.compile(r'([\\()])')
 
-
+@memoize
 def image_to_wd14_tags(image: Image.Image, model_name: str, threshold: float,
                        use_spaces: bool, use_escape: bool, include_ranks=False, score_descend=True) \
         -> Tuple[Mapping[str, float], str, Mapping[str, float]]:
@@ -139,12 +152,15 @@ def image_to_wd14_tags(image: Image.Image, model_name: str, threshold: float,
         tag_outformat = tag
         if use_spaces:
             tag_outformat = tag_outformat.replace('_', ' ')
+            tag_outformat.replace(' ', '-')
+        else:
+            tag_outformat.replace('_', '-')
         if use_escape:
             tag_outformat = re.sub(RE_SPECIAL, r'\\\1', tag_outformat)
         if include_ranks:
             tag_outformat = f"({tag_outformat}:{score:.3f})"
         text_items.append(tag_outformat)
-    output_text = ', '.join(text_items)
+    output_text = ' '.join(text_items)
 
     return ratings, output_text, filtered_tags