layout refine (#115)

2024-03-08 18:59:53 +08:00
parent fa171dfe6c
commit bcb58b7e71
5 changed files with 19 additions and 6 deletions
--- a/rag/app/naive.py
+++ b/rag/app/naive.py
@@ -101,7 +101,7 @@ def chunk(filename, binary=None, from_page=0, to_page=100000, lang="Chinese", ca
        d = copy.deepcopy(doc)
        if pdf_parser:
            d["image"], poss = pdf_parser.crop(ck, need_position=True)
-            add_positions(d, poss, from_page)
+            add_positions(d, poss)
            ck = pdf_parser.remove_tag(ck)
        tokenize(d, ck, eng)
        res.append(d)