Fix page range issue

VikParuchuri · Dec 3, 2024 · 619f5b0 · 619f5b0
1 parent 1e8d2f9
commit 619f5b0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/marker/builders/document.py b/marker/builders/document.py
@@ -37,10 +37,10 @@ def build_document(self, provider: PdfProvider):
         highres_images = provider.get_images(provider.page_range, self.highres_image_dpi)
         initial_pages = [
             PageGroupClass(
-                page_id=i,
+                page_id=p,
                 lowres_image=lowres_images[i],
                 highres_image=highres_images[i],
-                polygon=provider.get_page_bbox(i)
+                polygon=provider.get_page_bbox(p)
             ) for i, p in enumerate(provider.page_range)
         ]
         DocumentClass: Document = get_block_class(BlockTypes.Document)