|
|
@@ -147,7 +147,7 @@ def doc_analyze(
|
|
|
images.append(img_dict['img'])
|
|
|
page_wh_list.append((img_dict['width'], img_dict['height']))
|
|
|
|
|
|
- images_with_extra_info = [(images[index], ocr, dataset._lang) for index in range(len(dataset))]
|
|
|
+ images_with_extra_info = [(images[index], ocr, dataset._lang) for index in range(len(images))]
|
|
|
|
|
|
if len(images) >= MIN_BATCH_INFERENCE_SIZE:
|
|
|
batch_size = MIN_BATCH_INFERENCE_SIZE
|