Skip to content

Commit 48ef6c6

Browse files
committed
add batch prediction for layout model
Signed-off-by: SteffiVanHees <[email protected]>
1 parent 30f2e56 commit 48ef6c6

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

docling_ibm_models/layoutmodel/layout_predictor.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,6 @@ def predict(self, orig_img: Union[Image.Image, np.ndarray]) -> Iterable[dict]:
133133
page_img = Image.fromarray(orig_img).convert("RGB")
134134
else:
135135
raise TypeError("Not supported input image format")
136-
137136
resize = {"height": self._image_size, "width": self._image_size}
138137

139138
inputs = self._image_processor(

0 commit comments

Comments
 (0)