From c6813b052df5ba3df454860b11a8d77630e14edb Mon Sep 17 00:00:00 2001 From: Gabo Date: Fri, 18 Oct 2024 15:52:04 +0200 Subject: [PATCH] Omit formula errors --- src/extraction_formats/extract_formula_formats.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/extraction_formats/extract_formula_formats.py b/src/extraction_formats/extract_formula_formats.py index 70a05f3..154dc88 100644 --- a/src/extraction_formats/extract_formula_formats.py +++ b/src/extraction_formats/extract_formula_formats.py @@ -30,6 +30,6 @@ def extract_formula_format(pdf_images: PdfImages, predicted_segments: list[PdfSe formula_image = page_image.crop((left, top, left + width, top + height)) try: extracted_formula = get_latex_format(model, formula_image) - except RuntimeError: + except (ValueError, RuntimeError): continue predicted_segments[index].text_content = extracted_formula