Update app.py

app.py

@@ -81,10 +81,13 @@ def build_gradio_analyzer(table, table_ref, ocr):
     pipe_component_list.append(layout)
 
     if cfg.TAB:
-        cell = dd.SubImageLayoutService(d_cell, dd.LayoutType.table, {1: 6}, True)
+
+        detect_result_generator = DetectResultGenerator(categories_cell)
+        cell = SubImageLayoutService(d_cell, LayoutType.table, {1: 6}, detect_result_generator)
         pipe_component_list.append(cell)
 
-        item = dd.SubImageLayoutService(d_item, dd.LayoutType.table, {1: 7, 2: 8}, True)
+        detect_result_generator = DetectResultGenerator(categories_item)
+        item = SubImageLayoutService(d_item, LayoutType.table, {1: 7, 2: 8}, detect_result_generator)
         pipe_component_list.append(item)
 
         table_segmentation = dd.TableSegmentationService(