Upload search_image_by_image_or_text/pools.py with huggingface_hub

search_image_by_image_or_text/pools.py

@@ -0,0 +1,20 @@
+from typing import Type
+
+from cheesechaser.datapool import IncrementIDDataPool
+
+
+def quick_webp_pool(site_name: str, level: int = 3) -> Type[IncrementIDDataPool]:
+    repo_id = f'deepghs/{site_name}-webp-4Mpixel'
+
+    class _QuickWebpDataPool(IncrementIDDataPool):
+        def __init__(self, revision: str = 'main'):
+            IncrementIDDataPool.__init__(
+                self,
+                data_repo_id=repo_id,
+                data_revision=revision,
+                idx_repo_id=repo_id,
+                idx_revision=revision,
+                base_level=level,
+            )
+
+    return _QuickWebpDataPool