apunkt/PageIndex
1
0
Fork 0
mirror of https://github.com/VectifyAI/PageIndex.git synced 2026-04-24 23:56:21 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

fix model

Browse source
This commit is contained in:
Ray 2025-08-28 13:07:15 +08:00
parent 6d1b505541
commit 4b4ae4d51d
5 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

2
pageindex/utils.py
View file

@ -410,7 +410,7 @@ def add_preface_if_needed(data):
def get_page_tokens(pdf_path, model="gpt-4.1", pdf_parser="PyPDF2"):
def get_page_tokens(pdf_path, model="gpt-4o-2024-11-20", pdf_parser="PyPDF2"):
enc = tiktoken.encoding_for_model(model)
if pdf_parser == "PyPDF2":
pdf_reader = PyPDF2.PdfReader(pdf_path)