AutoPage / generate_page.sh
Mqleet's picture
upd code
fcaa164
#!/bin/bash
#qwen
# export QWEN_API_KEY='your_qwen_api_key' # openai
# export QWEN_API_BASE_URL='https://dashscope.aliyuncs.com/compatible-mode/v1' # openai
# deepseek
# export OPENAI_API_KEY='your_deepseek_api_key'
# export OPENAI_API_BASE_URL='https://api.deepseek.com'
# export OPENAI_API_KEY='your_openai_api_key' # openai
# export OPENAI_API_BASE_URL='https://api.openai.com' # openai
# gemini
# export GEMINI_API_KEY="your_gemini_api_key"
# ZhiPu glm
# export ZHIPUAI_API_KEY='your_glm_api_key'
# export OPENAI_API_BASE_URL='https://open.bigmodel.cn/api/paas/v4/'
export OPENROUTER_API_KEY="your_openrouter_api_key"
export OPENROUTER_API_BASE_URL="https://openrouter.ai/api/v1"
dataset_dir="pdfs"
paper_name="AutoPage"
python -m ProjectPageAgent.main_pipline\
--paper_path="${dataset_dir}/${paper_name}.pdf" \
--model_name_t="your_text_model" \
--model_name_v="your_vlm_model" \
--template_root="templates" \
--template_dir="your_template_dir" \
--template_file="your_template_file" \
--output_dir="generated_project_pages" \
--full_content_check_times=2 \
--html_check_times=2 \
--resume='parse_pdf' \
--human_input='1' \
--background_color='dark' \
--has_navigation="yes" \
--has_hero_section="no" \
--title_color="colorful" \
--page_density="compact" \
--image_layout="rotation"