diff --git a/CHANGELOG.md b/CHANGELOG.md index e054dfee..b10a7d88 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,4 +1,4 @@ -## 0.7.3-dev2 +## 0.7.3 * Integration of Chipperv2 and additional Chipper functionality, which includes automatic detection of GPU, bounding box prediction and hierarchical representation. diff --git a/test_unstructured_inference/models/test_chippermodel.py b/test_unstructured_inference/models/test_chippermodel.py index 17840d03..a0b1a6c1 100644 --- a/test_unstructured_inference/models/test_chippermodel.py +++ b/test_unstructured_inference/models/test_chippermodel.py @@ -237,7 +237,7 @@ def test_postprocess_bbox(decoded_str, expected_classes): def test_run_chipper_v2(): - model = get_model("chipperv2") + model = get_model("chipper") img = Image.open("sample-docs/easy_table.jpg") elements = model(img) tables = [el for el in elements if el.type == "Table"] diff --git a/unstructured_inference/__version__.py b/unstructured_inference/__version__.py index 7e5ecedb..26d31cde 100644 --- a/unstructured_inference/__version__.py +++ b/unstructured_inference/__version__.py @@ -1 +1 @@ -__version__ = "0.7.3-dev2" # pragma: no cover +__version__ = "0.7.3" # pragma: no cover diff --git a/unstructured_inference/constants.py b/unstructured_inference/constants.py index 3585b077..e13e64d1 100644 --- a/unstructured_inference/constants.py +++ b/unstructured_inference/constants.py @@ -11,6 +11,7 @@ class Source(Enum): DETECTRON2_ONNX = "detectron2_onnx" DETECTRON2_LP = "detectron2_lp" CHIPPER = "chipper" + CHIPPERV1 = "chipperv1" CHIPPERV2 = "chipperv2" PDFMINER = "pdfminer" MERGED = "merged" diff --git a/unstructured_inference/models/chipper.py b/unstructured_inference/models/chipper.py index 3a1d04ea..315b7c57 100644 --- a/unstructured_inference/models/chipper.py +++ b/unstructured_inference/models/chipper.py @@ -18,7 +18,7 @@ from unstructured_inference.utils import LazyDict MODEL_TYPES: Dict[Optional[str], Union[LazyDict, dict]] = { - "chipper": { + "chipperv1": { "pre_trained_model_repo": "unstructuredio/ved-fine-tuning", "swap_head": False, "start_token_prefix": " List[LayoutElement]: """For chipper, remove elements from other sources.""" - return [el for el in elements if el.source in (Source.CHIPPER, Source.CHIPPERV2)] + return [el for el in elements if el.source in (Source.CHIPPER, Source.CHIPPERV1)] def adjust_bbox(self, bbox, x_offset, y_offset, ratio, target_size): """Translate bbox by (x_offset, y_offset) and shrink by ratio."""