chore: update run pipline script to use all parameters that are used in production
This commit is contained in:
parent
9ff88a1e5d
commit
bf9ab4b1a2
@ -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]
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user