add a new demo

CosmosShadow · Jul 5, 2024 · 8a21c8f · 8a21c8f
1 parent 5cef400
commit 8a21c8f
Show file tree

Hide file tree

Showing 12 changed files with 335 additions and 1 deletion.
diff --git a/examples/rh.pdf b/examples/rh.pdf
diff --git a/examples/rh/0_0.png b/examples/rh/0_0.png
diff --git a/examples/rh/0_1.png b/examples/rh/0_1.png
diff --git a/examples/rh/2_0.png b/examples/rh/2_0.png
diff --git a/examples/rh/5_0.png b/examples/rh/5_0.png
diff --git a/examples/rh/5_1.png b/examples/rh/5_1.png
diff --git a/examples/rh/6_0.png b/examples/rh/6_0.png
diff --git a/examples/rh/7_0.png b/examples/rh/7_0.png
diff --git a/examples/rh/8_0.png b/examples/rh/8_0.png
diff --git a/examples/rh/9_0.png b/examples/rh/9_0.png
diff --git a/examples/rh/output.md b/examples/rh/output.md
diff --git a/test/test.py b/test/test.py
@@ -7,6 +7,9 @@
 pdf_path = '../examples/attention_is_all_you_need.pdf'
 output_dir = '../examples/attention_is_all_you_need/'   
 
+pdf_path = '../examples/rh.pdf'
+output_dir = '../examples/rh/'
+
 # 清空output_dir
 import shutil
 shutil.rmtree(output_dir, ignore_errors=True)
@@ -24,13 +27,13 @@ def test_use_api_key():
 
 def test_use_env():
     from gptpdf import parse_pdf
-
     # Use OPENAI_API_KEY and OPENAI_API_BASE from environment variables
     content, image_paths = parse_pdf(pdf_path, output_dir=output_dir, model='gpt-4o', verbose=True)
     print(content)
     print(image_paths)
     # also output_dir/output.md is generated
 
+
 def test_azure():
     from gptpdf import parse_pdf
     api_key = '8ef0b4df45e444079cd5a4xxxxx' # Azure API Key