fix: image save take 2

andreiionutdamian · Apr 23, 2024 · 0d9aa98 · 0d9aa98
1 parent 91ef12e
commit 0d9aa98
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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")