Enable Semantic Search on Images

2026-03-03 13:19:16 +00:00 · 2021-08-22 21:00:54 -07:00
parent fd217fe8b7
commit 4daeddbbda
5 changed files with 104 additions and 28 deletions
--- a/src/utils/cli.py
+++ b/src/utils/cli.py
@@ -53,6 +53,10 @@ default_config = {
        {
            'compressed-jsonl': '.transactions.jsonl.gz',
            'embeddings-file': '.transaction_embeddings.pt'
+        },
+        'image':
+        {
+            'embeddings-file': '.image_embeddings.pt'
        }
    },
    'search-type':
@@ -61,6 +65,10 @@ default_config = {
        {
            'encoder': "sentence-transformers/msmarco-MiniLM-L-6-v3",
            'cross-encoder': "cross-encoder/ms-marco-MiniLM-L-6-v2"
+        },
+        'image':
+        {
+            'encoder': "clip-ViT-B-32"
        }
    }
 }