From 0d9aa98621622651c3b706fca4464d727e2815f5 Mon Sep 17 00:00:00 2001 From: ispirtraian Date: Tue, 23 Apr 2024 15:37:06 +0000 Subject: [PATCH] fix: image save take 2 --- app/src/serving/serving_app.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/app/src/serving/serving_app.py b/app/src/serving/serving_app.py index 5f190ec..74895ac 100644 --- a/app/src/serving/serving_app.py +++ b/app/src/serving/serving_app.py @@ -344,8 +344,7 @@ def _predict_job (self, model_type: str, input, params:dict = None): with open(task_content_path, 'wb') as file: file.write(content.encode('utf-8')) elif isinstance(content, Image.Image): - with content as img: - img.save(task_content_path, format=content.format) + content.save(task_content_path, format=content.format) else: raise Exception("Unsupported content type")