add additional args to TableSegmentationService

app.py

@@ -150,6 +150,10 @@ def build_gradio_analyzer():
             cfg.SEGMENTATION.FULL_TABLE_TILING,
             cfg.SEGMENTATION.REMOVE_IOU_THRESHOLD_ROWS,
             cfg.SEGMENTATION.REMOVE_IOU_THRESHOLD_COLS,
+            dd.LayoutType.table,
+            [dd.CellType.header, dd.CellType.body, dd.LayoutType.cell],
+            [dd.LayoutType.row, dd.LayoutType.column],
+            [dd.CellType.row_number, dd.CellType.column_number],
             cfg.SEGMENTATION.STRETCH_RULE
         )
         pipe_component_list.append(table_segmentation)