refine page ranges (#147)

2024-03-25 13:11:57 +08:00
parent 1d9a50b090
commit 71fe314955
13 changed files with 169 additions and 72 deletions
--- a/rag/app/paper.py
+++ b/rag/app/paper.py
@@ -136,7 +136,7 @@ def chunk(filename, binary=None, from_page=0, to_page=100000, lang="Chinese", ca
                "title": filename,
                "authors": " ",
                "abstract": "",
-                "sections": pdf_parser(filename if not binary else binary),
+                "sections": pdf_parser(filename if not binary else binary, from_page=from_page, to_page=to_page),
                "tables": []
            }
        else: