diff --git a/scripts/run_pipeline.py b/scripts/run_pipeline.py
index c2b4bb0..f12fb1d 100644
--- a/scripts/run_pipeline.py
+++ b/scripts/run_pipeline.py
@@ -3,6 +3,7 @@ import json
 import os
 from glob import glob
 
+from image_prediction.config import CONFIG
 from image_prediction.pipeline import load_pipeline
 from image_prediction.utils import get_logger
 from image_prediction.utils.pdf_annotation import annotate_pdf
@@ -35,7 +36,7 @@ def process_pdf(pipeline, pdf_path, page_range=None):
 
 
 def main(args):
-    pipeline = load_pipeline(verbose=True, tolerance=3)
+    pipeline = load_pipeline(verbose=True, batch_size=CONFIG.service.batch_size)
 
     if os.path.isfile(args.input):
         pdf_paths = [args.input]