浏览代码

fix doc_analyze first page only

Nathan Dahlberg 8 月之前
父节点
当前提交
37f3e200da
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      magic_pdf/model/doc_analyze_by_custom_model.py

+ 1 - 1
magic_pdf/model/doc_analyze_by_custom_model.py

@@ -158,7 +158,7 @@ def doc_analyze(
     table_enable=None,
     table_enable=None,
 ) -> InferenceResult:
 ) -> 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()
     model_manager = ModelSingleton()
     custom_model = model_manager.get_model(
     custom_model = model_manager.get_model(