runner: Accept truncated images in routes

livepeer · Feb 29, 2024 · 62931de · 62931de
1 parent 0defbac
commit 62931de
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/runner/app/routes/image_to_image.py b/runner/app/routes/image_to_image.py
@@ -10,6 +10,10 @@
 import random
 import os
 
+from PIL import ImageFile
+
+ImageFile.LOAD_TRUNCATED_IMAGES = True
+
 router = APIRouter()
 
 logger = logging.getLogger(__name__)

diff --git a/runner/app/routes/image_to_video.py b/runner/app/routes/image_to_video.py
@@ -10,6 +10,10 @@
 import random
 import os
 
+from PIL import ImageFile
+
+ImageFile.LOAD_TRUNCATED_IMAGES = True
+
 router = APIRouter()
 
 logger = logging.getLogger(__name__)