fix: image format

Signed-off-by: Frost Ming <[email protected]>
bentoml · Feb 1, 2024 · 6160d05 · 6160d05
1 parent bde63b7
commit 6160d05
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/_bentoml_sdk/io_models.py b/src/_bentoml_sdk/io_models.py
@@ -188,10 +188,11 @@ def content_stream() -> t.Generator[str | bytes, None, None]:
                     )
                 else:  # is PIL Image
                     buffer = io.BytesIO()
-                    obj.save(buffer, format=obj.format)
+                    image_format = obj.format or "PNG"
+                    obj.save(buffer, format=image_format)
                     return Response(
                         content=buffer.getvalue(),
-                        media_type=f"image/{obj.format.lower()}",
+                        media_type=f"image/{image_format.lower()}",
                     )
 
             if not isinstance(obj, RootModel):