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__)