mirror of
https://github.com/VectifyAI/PageIndex.git
synced 2026-04-24 23:56:21 +02:00
fix option for adding node text
This commit is contained in:
parent
36b22e3bc7
commit
ad65e3f19c
2 changed files with 10 additions and 5 deletions
|
|
@ -18,6 +18,8 @@ if __name__ == "__main__":
|
|||
help='Whether to add summary to the node')
|
||||
parser.add_argument('--if-add-doc-description', type=str, default='yes',
|
||||
help='Whether to add doc description to the doc')
|
||||
parser.add_argument('--if-add-node-text', type=str, default='no',
|
||||
help='Whether to add text to the node')
|
||||
args = parser.parse_args()
|
||||
|
||||
# Configure options
|
||||
|
|
@ -28,7 +30,8 @@ if __name__ == "__main__":
|
|||
max_token_num_each_node=args.max_tokens_per_node,
|
||||
if_add_node_id=args.if_add_node_id,
|
||||
if_add_node_summary=args.if_add_node_summary,
|
||||
if_add_doc_description=args.if_add_doc_description
|
||||
if_add_doc_description=args.if_add_doc_description,
|
||||
if_add_node_text=args.if_add_node_text
|
||||
)
|
||||
|
||||
# Process the PDF
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue