Browse Source

is_pretty_markdown -> pretty_markdown

gaotingquan 6 tháng trước cách đây
mục cha
commit
319e7fac22

+ 5 - 5
paddlex/inference/pipelines/layout_parsing/pipeline_v2.py

@@ -896,7 +896,7 @@ class LayoutParsingPipelineV2(BasePipeline):
         use_formula_recognition: Union[bool, None],
         use_chart_recognition: Union[bool, None],
         use_region_detection: Union[bool, None],
-        is_pretty_markdown: Union[bool, None],
+        pretty_markdown: Union[bool, None],
     ) -> dict:
         """
         Get the model settings based on the provided parameters or default values.
@@ -947,7 +947,7 @@ class LayoutParsingPipelineV2(BasePipeline):
             use_formula_recognition=use_formula_recognition,
             use_chart_recognition=use_chart_recognition,
             use_region_detection=use_region_detection,
-            is_pretty_markdown=is_pretty_markdown,
+            pretty_markdown=pretty_markdown,
         )
 
     def predict(
@@ -983,7 +983,7 @@ class LayoutParsingPipelineV2(BasePipeline):
         use_e2e_wireless_table_rec_model: bool = True,
         max_new_tokens: int = 1024,
         no_repeat_ngram_size: int = 20,
-        is_pretty_markdown: Union[bool, None] = None,
+        pretty_markdown: Union[bool, None] = None,
         **kwargs,
     ) -> LayoutParsingResultV2:
         """
@@ -1023,7 +1023,7 @@ class LayoutParsingPipelineV2(BasePipeline):
             use_e2e_wireless_table_rec_model (bool): Whether to use end-to-end wireless table recognition model.
             max_new_tokens: int = 1024,
             no_repeat_ngram_size: int = 20,
-            is_pretty_markdown,
+            pretty_markdown,
             **kwargs (Any): Additional settings to extend functionality.
 
         Returns:
@@ -1039,7 +1039,7 @@ class LayoutParsingPipelineV2(BasePipeline):
             use_formula_recognition,
             use_chart_recognition,
             use_region_detection,
-            is_pretty_markdown,
+            pretty_markdown,
         )
 
         if not self.check_model_settings_valid(model_settings):

+ 1 - 1
paddlex/inference/pipelines/layout_parsing/result_v2.py

@@ -407,7 +407,7 @@ class LayoutParsingResultV2(BaseCVResult, HtmlMixin, XlsxMixin, MarkdownMixin):
 
                 return seg_start_flag, seg_end_flag
 
-            if self["model_settings"].get("is_pretty_markdown", True):
+            if self["model_settings"].get("pretty_markdown", True):
                 format_text = format_text_centered_by_html
                 format_image = format_image_centered_by_html
             else: