Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

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

Browse Source
This commit is contained in:
Julius Unverfehrt 2024-08-30 10:30:04 +02:00
parent 9ff88a1e5d
commit bf9ab4b1a2
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/run_pipeline.py
View File

@ -3,6 +3,7 @@ import json
import os import os
from glob import glob from glob import glob
from image_prediction.config import CONFIG
from image_prediction.pipeline import load_pipeline from image_prediction.pipeline import load_pipeline
from image_prediction.utils import get_logger from image_prediction.utils import get_logger
from image_prediction.utils.pdf_annotation import annotate_pdf 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): 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): if os.path.isfile(args.input):
pdf_paths = [args.input] pdf_paths = [args.input]