From ad8e831a5cbe5bb65d0567d4bdde95bd3ef3de75 Mon Sep 17 00:00:00 2001 From: Lia Lenckowski Date: Thu, 7 Sep 2023 16:41:02 +0200 Subject: add progress bars/spinners --- src/imgbeddings-api.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) (limited to 'src/imgbeddings-api.py') diff --git a/src/imgbeddings-api.py b/src/imgbeddings-api.py index 795c625..0c890b5 100644 --- a/src/imgbeddings-api.py +++ b/src/imgbeddings-api.py @@ -1,15 +1,18 @@ from PIL import Image from imgbeddings import imgbeddings -import sys +import sys, os import json as j -#from itertools import batched # TODO das hier ab python 3.12 b = imgbeddings() -ems = [] +paths = sys.argv[1:] +batch_size = 8 -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)] +for i in range(0, len(paths), batch_size): + fs = paths[i:i+batch_size] -print(j.dumps([em[0].tolist() for em in ems])) + 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") -- cgit v1.2.3-70-g09d2