Home Explore Help
Register Sign In
zhengchun
/
MinerU
1
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Browse Source

fix doc_analyze first page only

Nathan Dahlberg 8 months ago
parent
c4849ba0e6
commit
37f3e200da
1 changed files with 1 additions and 1 deletions
Split View Show Diff Stats
  1. 1 1
      magic_pdf/model/doc_analyze_by_custom_model.py

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

@@ -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(