pass in detection batch size appropriately for ocr_extraction [skip ci]

VikParuchuri · Nov 21, 2024 · b98e0a3 · b98e0a3
1 parent b721d11
commit b98e0a3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/marker/builders/ocr.py b/marker/builders/ocr.py
@@ -71,7 +71,8 @@ def ocr_extraction(self, document: Document, provider: PdfProvider) -> ProviderP
             det_processor=self.detection_model.processor,
             rec_model=self.recognition_model,
             rec_processor=self.recognition_model.processor,
-            batch_size=int(self.get_recognition_batch_size()),
+            detection_batch_size=int(self.get_detection_batch_size()),
+            recognition_batch_size=int(self.get_recognition_batch_size()),
             highres_images=[page.highres_image for page in page_list]
         )