Skip to content

Commit e402cd2

Browse files
committed
chore: update demo
1 parent fc4c1b9 commit e402cd2

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

demo.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,7 @@
1010
input_args = RapidTableInput(model_type=ModelType.UNITABLE)
1111
table_engine = RapidTable(input_args)
1212

13-
img_path = "tests/test_files/table_without_txt.jpg"
14-
# img_path = "https://raw.githubusercontent.com/RapidAI/RapidTable/refs/heads/main/tests/test_files/table.jpg"
13+
img_path = "https://raw.githubusercontent.com/RapidAI/RapidTable/refs/heads/main/tests/test_files/table.jpg"
1514

1615
# # 使用单字识别
1716
# ori_ocr_res = ocr_engine(img_path, return_word_box=True)
@@ -23,5 +22,5 @@
2322

2423
ori_ocr_res = ocr_engine(img_path)
2524
ocr_results = [ori_ocr_res.boxes, ori_ocr_res.txts, ori_ocr_res.scores]
26-
results = table_engine(img_path)
25+
results = table_engine(img_path, ocr_results=ocr_results)
2726
results.vis(save_dir="outputs", save_name="vis")

0 commit comments

Comments
 (0)