Omit formula errors

huridocs · Oct 18, 2024 · c6813b0 · c6813b0
1 parent 20b325f
commit c6813b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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