Fix decompression bomb error

VikParuchuri · Dec 3, 2024 · 91862fb · 91862fb
1 parent ee96b54
commit 91862fb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/marker/renderers/html.py b/marker/renderers/html.py
@@ -1,4 +1,3 @@
-import re
 from typing import Literal
 
 from bs4 import BeautifulSoup, MarkupResemblesLocatorWarning
@@ -7,14 +6,16 @@
 from marker.renderers import BaseRenderer
 from marker.schema import BlockTypes
 from marker.schema.blocks import BlockId
+from marker.settings import settings
 
 # Ignore beautifulsoup warnings
 import warnings
-
-from marker.settings import settings
-
 warnings.filterwarnings("ignore", category=MarkupResemblesLocatorWarning)
 
+# Suppress DecompressionBombError
+from PIL import Image
+Image.MAX_IMAGE_PIXELS = None
+
 
 class HTMLOutput(BaseModel):
     html: str

diff --git a/marker/settings.py b/marker/settings.py
@@ -1,4 +1,4 @@
-from typing import Optional, List, Dict, Literal
+from typing import Optional
 
 from dotenv import find_dotenv
 from pydantic import computed_field