瀏覽代碼

refactor: close PDF document after processing to improve resource management

myhloli 5 月之前
父節點
當前提交
3ff994c45e
共有 1 個文件被更改,包括 2 次插入0 次删除
  1. 2 0
      mineru/backend/vlm/token_to_middle_json.py

+ 2 - 0
mineru/backend/vlm/token_to_middle_json.py

@@ -48,6 +48,8 @@ def result_to_middle_json(token_list, images_list, pdf_doc, image_writer):
         image_dict = images_list[index]
         page_info = token_to_page_info(token, image_dict, page, image_writer, index)
         middle_json["pdf_info"].append(page_info)
+    # 关闭pdf文档
+    pdf_doc.close()
     return middle_json