diff --git a/src/eynollah/model_zoo/model_zoo.py b/src/eynollah/model_zoo/model_zoo.py
index 0dd24a8..49ed8e1 100644
--- a/src/eynollah/model_zoo/model_zoo.py
+++ b/src/eynollah/model_zoo/model_zoo.py
@@ -379,9 +379,9 @@ class EynollahModelZoo:
             return text, conf
         model.predict_on_batch = predict_torch
         # not actually needed (image processor does resize itself)
+        # no batch dimension (images passed as list w/ varying shapes)
         model.input_shape = (None,
-                             proc.image_processor.size.height,
-                             proc.image_processor.size.width,
+                             None,
                              len(proc.image_processor.image_mean))
         return model