From c4b03717914e5c907f7f47dc2a85df6b57763c58 Mon Sep 17 00:00:00 2001 From: Lia Lenckowski Date: Thu, 7 Sep 2023 01:37:34 +0200 Subject: add content embedder --- src/imgbeddings-api.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 src/imgbeddings-api.py (limited to 'src/imgbeddings-api.py') diff --git a/src/imgbeddings-api.py b/src/imgbeddings-api.py new file mode 100644 index 0000000..795c625 --- /dev/null +++ b/src/imgbeddings-api.py @@ -0,0 +1,15 @@ +from PIL import Image +from imgbeddings import imgbeddings +import sys +import json as j +#from itertools import batched # TODO das hier ab python 3.12 + +b = imgbeddings() + +ems = [] + +for f in sys.argv[1:]: # TODO this should be batched for faster ai stuff + im = Image.open(open(f, "rb")) + ems += [b.to_embeddings(im)] + +print(j.dumps([em[0].tolist() for em in ems])) -- cgit v1.2.3-70-g09d2