diff options
| author | Lia Lenckowski <lialenck@protonmail.com> | 2024-11-26 19:11:08 +0100 |
|---|---|---|
| committer | Lia Lenckowski <lialenck@protonmail.com> | 2024-11-26 19:11:08 +0100 |
| commit | 58c1da21ec0e7a2a52cdc84469fe05d25bd94e13 (patch) | |
| tree | edbb8e10f112e9e89875b5ac71bc93ae763d2f17 /src/embedders/imgbeddings-api.py | |
| parent | 6a2d5e2241b73039ae09b627c30841248d887a79 (diff) | |
| download | embeddings-sort-58c1da21ec0e7a2a52cdc84469fe05d25bd94e13.tar embeddings-sort-58c1da21ec0e7a2a52cdc84469fe05d25bd94e13.tar.bz2 embeddings-sort-58c1da21ec0e7a2a52cdc84469fe05d25bd94e13.tar.zst | |
make ai embedding more resilient, remove dependency on python
Diffstat (limited to 'src/embedders/imgbeddings-api.py')
| -rw-r--r-- | src/embedders/imgbeddings-api.py | 18 |
1 files changed, 0 insertions, 18 deletions
diff --git a/src/embedders/imgbeddings-api.py b/src/embedders/imgbeddings-api.py deleted file mode 100644 index 0c890b5..0000000 --- a/src/embedders/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") |