fix doc_analyze first page only

37f3e200 · Nathan Dahlberg · GitHub · c4849ba0 · 37f3e200
Unverified Commit 37f3e200 authored Feb 22, 2025 by Nathan Dahlberg Committed by GitHub Feb 22, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

magic_pdf/model/doc_analyze_by_custom_model.py magic_pdf/model/doc_analyze_by_custom_model.py +1 -1

No files found.
--- a/magic_pdf/model/doc_analyze_by_custom_model.py
+++ b/magic_pdf/model/doc_analyze_by_custom_model.py
@@ -158,7 +158,7 @@ def doc_analyze(
    table_enable=None,
 ) -> InferenceResult:

-    end_page_id = end_page_id if end_page_id else len(dataset) - 1
+    end_page_id = end_page_id if end_page_id is not None else len(dataset) - 1

    model_manager = ModelSingleton()
    custom_model = model_manager.get_model(