From bf9ab4b1a2da0c538954b73b6ad933481b85c26b Mon Sep 17 00:00:00 2001 From: Julius Unverfehrt Date: Fri, 30 Aug 2024 10:30:04 +0200 Subject: [PATCH] chore: update run pipline script to use all parameters that are used in production --- scripts/run_pipeline.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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]