Commit ab93f856 authored by myhloli's avatar myhloli
Browse files

refactor: add 'local' option for model source and comment out timing logs

parent 29d26261
...@@ -47,11 +47,11 @@ def doc_analyze( ...@@ -47,11 +47,11 @@ def doc_analyze(
if predictor is None: if predictor is None:
predictor = ModelSingleton().get_model(backend, model_path, server_url) predictor = ModelSingleton().get_model(backend, model_path, server_url)
load_images_start = time.time() # load_images_start = time.time()
images_list, pdf_doc = load_images_from_pdf(pdf_bytes) images_list, pdf_doc = load_images_from_pdf(pdf_bytes)
images_base64_list = [image_dict["img_base64"] for image_dict in images_list] images_base64_list = [image_dict["img_base64"] for image_dict in images_list]
load_images_time = round(time.time() - load_images_start, 2) # load_images_time = round(time.time() - load_images_start, 2)
logger.info(f"load images cost: {load_images_time}, speed: {round(len(images_base64_list)/load_images_time, 3)} images/s") # logger.info(f"load images cost: {load_images_time}, speed: {round(len(images_base64_list)/load_images_time, 3)} images/s")
infer_start = time.time() infer_start = time.time()
results = predictor.batch_predict(images=images_base64_list) results = predictor.batch_predict(images=images_base64_list)
......
...@@ -129,7 +129,7 @@ from .common import do_parse, read_fn, pdf_suffixes, image_suffixes ...@@ -129,7 +129,7 @@ from .common import do_parse, read_fn, pdf_suffixes, image_suffixes
'-r', '-r',
'--repo', '--repo',
'model_source', 'model_source',
type=click.Choice(['huggingface', 'modelscope']), type=click.Choice(['huggingface', 'modelscope', 'local']),
help=""" help="""
The source of the model repository. Default is 'huggingface'. The source of the model repository. Default is 'huggingface'.
""", """,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment