瀏覽代碼

Fix SLANet_plus infer bug (#2871)

* add table models

* refine codes

* refine codes

* fix bugs
Liu Jiaxuan 10 月之前
父節點
當前提交
47cda8aaac
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/inference/models_new/table_structure_recognition/predictor.py

+ 1 - 1
paddlex/inference/models_new/table_structure_recognition/predictor.py

@@ -66,7 +66,7 @@ class TablePredictor(BasicPredictor):
         )
 
         postprocessors = TableLabelDecode(
-            model_name="SLANet",
+            model_name=self.config["Global"]["model_name"],
             merge_no_span_structure=self.config["PreProcess"]["transform_ops"][1][
                 "TableLabelEncode"
             ]["merge_no_span_structure"],