From 42a7d792c30f4d5ec08a7f39351145303100d318 Mon Sep 17 00:00:00 2001 From: myhloli Date: Sat, 28 Sep 2024 23:16:18 +0800 Subject: [PATCH] refactor(magic_pdf): import model helpers directly for clarity Update import statements in `pdf_parse_union_core_v2.py` to directly import `prepare_inputs`, `boxes2inputs`, and `parse_logits` from `magic_pdf.model.v3.helpers` instead of from `magic_pdf.model.v3`. This change streamlines the imports, making the code more readable and maintaining a cleaner approach to modular design. --- magic_pdf/pdf_parse_union_core_v2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/magic_pdf/pdf_parse_union_core_v2.py b/magic_pdf/pdf_parse_union_core_v2.py index 2123a34d..60bb92e5 100644 --- a/magic_pdf/pdf_parse_union_core_v2.py +++ b/magic_pdf/pdf_parse_union_core_v2.py @@ -140,7 +140,7 @@ def get_model(self, model_name: str, local_path=None): def do_predict(boxes: List[List[int]], model) -> List[int]: - from magic_pdf.model.v3 import prepare_inputs, boxes2inputs, parse_logits + from magic_pdf.model.v3.helpers import prepare_inputs, boxes2inputs, parse_logits inputs = boxes2inputs(boxes) inputs = prepare_inputs(inputs, model) logits = model(**inputs).logits.cpu().squeeze(0)