chore: update run pipline script to use all parameters that are used in production

This commit is contained in:
Julius Unverfehrt 2024-08-30 10:30:04 +02:00
parent 9ff88a1e5d
commit bf9ab4b1a2

View File

@ -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]