From 62931def1de359250e683cfc6019c09101edd4d3 Mon Sep 17 00:00:00 2001
From: Yondon Fu <yondon.fu@gmail.com>
Date: Thu, 29 Feb 2024 18:38:18 -0500
Subject: [PATCH] runner: Accept truncated images in routes

---
 runner/app/routes/image_to_image.py | 4 ++++
 runner/app/routes/image_to_video.py | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/runner/app/routes/image_to_image.py b/runner/app/routes/image_to_image.py
index a1df07a..6af61bc 100644
--- 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
index 99a9870..f61e699 100644
--- 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__)