Update pdf_extract_kit.py

magic_pdf/model/pdf_extract_kit.py

@@ -370,7 +370,7 @@ class CustomPEKModel:
 
             ocr_cost = round(time.time() - ocr_start, 2)
             # logger.info(f"ocr cost: {ocr_cost}")
-            total_cost = total_cost + ocr_cost
+            total_cost = round(total_cost + ocr_cost,2)
         index = index + 1
         end_page_id = end_page_id + 1
         logger.info(f'当前解析第【{index} / {end_page_id}】页, 耗时：{total_cost}')