zhouchangda 1 年之前
父节点
当前提交
d6a2dbcd99
共有 2 个文件被更改,包括 4 次插入4 次删除
  1. 1 1
      paddlex/inference/pipelines/ppchatocrv3/ppchatocrv3.py
  2. 3 3
      paddlex/inference/results/table_rec.py

+ 1 - 1
paddlex/inference/pipelines/ppchatocrv3/ppchatocrv3.py

@@ -571,7 +571,7 @@ class PPChatOCRPipeline(_TableRecPipeline):
             if res:
             if res:
                 final_results.update(res)
                 final_results.update(res)
         if not res and not final_results:
         if not res and not final_results:
-            final_results = llm_api.ERROR_MASSAGE
+            final_results = {"error": llm_api.ERROR_MASSAGE}
         if save_prompt:
         if save_prompt:
             return ChatResult({"chat_res": final_results, "prompt": prompt_res})
             return ChatResult({"chat_res": final_results, "prompt": prompt_res})
         else:
         else:

+ 3 - 3
paddlex/inference/results/table_rec.py

@@ -100,10 +100,10 @@ class TableResult(BaseResult):
             save_path = Path(save_path).stem
             save_path = Path(save_path).stem
         layout_save_path = f"{save_path}_layout.jpg"
         layout_save_path = f"{save_path}_layout.jpg"
         ocr_save_path = f"{save_path}_ocr.jpg"
         ocr_save_path = f"{save_path}_ocr.jpg"
-        table_save_path = f"{save_path}_table.jpg"
+        table_save_path = f"{save_path}_table"
         layout_result = self["layout_result"]
         layout_result = self["layout_result"]
         layout_result.save_to_img(layout_save_path)
         layout_result.save_to_img(layout_save_path)
         ocr_result = self["ocr_result"]
         ocr_result = self["ocr_result"]
         ocr_result.save_to_img(ocr_save_path)
         ocr_result.save_to_img(ocr_save_path)
-        for table_result in self["table_result"]:
-            table_result.save_to_img(table_save_path)
+        for idx, table_result in enumerate(self["table_result"]):
+            table_result.save_to_img(f"{table_save_path}_{idx}.jpg")