Add table dt

main.py

@@ -26,7 +26,12 @@ def get_rec(points):
         
     
 def predict(image_input):
-    return ocr(image_input)
+    ocr_results = ocr(image_input)
+    table_results = inference_detector(table_det, image_input)
+    table_bboxes_list = [bboxes for bboxes, rois in table_results]
+    table_rois_list = [rois for bboxes, rois in table_results]
+
+    return {**ocr_results, 'table_bboxes': table_bboxes_list, 'table_rois': table_rois_list}
 
 def run():
     demo = gr.Interface(