1 files changed, 10 insertions, 7 deletions
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")