Browse Source

Merge pull request #1753 from myhloli/dev

fix(llm_aided): update prompt
Xiaomeng Zhao 9 months ago
parent
commit
9fd10b66e4

+ 5 - 1
magic_pdf/model/doc_analyze_by_custom_model.py

@@ -145,7 +145,11 @@ def doc_analyze(
     table_enable=None,
 ) -> InferenceResult:
 
-    end_page_id = end_page_id if end_page_id is not None else len(dataset) - 1
+    end_page_id = (
+        end_page_id
+        if end_page_id is not None and end_page_id >= 0
+        else len(dataset) - 1
+    )
 
     model_manager = ModelSingleton()
     custom_model = model_manager.get_model(

+ 1 - 1
magic_pdf/post_proc/llm_aided.py

@@ -120,7 +120,7 @@ def llm_aided_title(pdf_info_dict, title_aided_config):
     - 在完成初步分级后,仔细检查分级结果的合理性
     - 根据上下文关系和逻辑顺序,对不合理的分级进行微调
     - 确保最终的分级结果符合文档的实际结构和逻辑
-    - 字典中包含OCR错误识别的标题,你可以通过将其层级标记为 0 来排除它们
+    - 字典中可能包含被误当成标题的正文,你可以通过将其层级标记为 0 来排除它们
     
 IMPORTANT: 
 请直接返回优化过的由标题层级组成的字典,格式为{{标题id:标题层级}},如下: