Explorar o código

refactor: close PDF document after processing to improve resource management

myhloli hai 5 meses
pai
achega
3ff994c45e
Modificáronse 1 ficheiros con 2 adicións e 0 borrados
  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]
         image_dict = images_list[index]
         page_info = token_to_page_info(token, image_dict, page, image_writer, index)
         page_info = token_to_page_info(token, image_dict, page, image_writer, index)
         middle_json["pdf_info"].append(page_info)
         middle_json["pdf_info"].append(page_info)
+    # 关闭pdf文档
+    pdf_doc.close()
     return middle_json
     return middle_json