diff options
| author | metamuffin <metamuffin@disroot.org> | 2023-09-20 16:55:50 +0200 |
|---|---|---|
| committer | metamuffin <metamuffin@disroot.org> | 2023-09-20 16:55:50 +0200 |
| commit | fbfee0a2bb436a6205d67f561dbd6284621504d6 (patch) | |
| tree | 8bc973c19f1e3ee12eaa382fa63a20ddf642fdfe /src/imgbeddings-api.py | |
| parent | f62b4e356a1deecc550a2eba6d7d0caaad1303c1 (diff) | |
| download | embeddings-sort-fbfee0a2bb436a6205d67f561dbd6284621504d6.tar embeddings-sort-fbfee0a2bb436a6205d67f561dbd6284621504d6.tar.bz2 embeddings-sort-fbfee0a2bb436a6205d67f561dbd6284621504d6.tar.zst | |
move embedder to module
Diffstat (limited to 'src/imgbeddings-api.py')
| -rw-r--r-- | src/imgbeddings-api.py | 18 |
1 files changed, 0 insertions, 18 deletions
diff --git a/src/imgbeddings-api.py b/src/imgbeddings-api.py deleted file mode 100644 index 0c890b5..0000000 --- a/src/imgbeddings-api.py +++ /dev/null @@ -1,18 +0,0 @@ -from PIL import Image -from imgbeddings import imgbeddings -import sys, os -import json as j - -b = imgbeddings() - -paths = sys.argv[1:] -batch_size = 8 - -for i in range(0, len(paths), batch_size): - fs = paths[i:i+batch_size] - - ims = [Image.open(open(f, "rb")) for f in fs] - for emb in b.to_embeddings(ims).tolist(): - print(j.dumps(emb)) - -sys.stderr.write("\n") |