diff --git a/backend/open_webui/routers/models.py b/backend/open_webui/routers/models.py index 4475b2d78e..c5d8b6d102 100644 --- a/backend/open_webui/routers/models.py +++ b/backend/open_webui/routers/models.py @@ -306,12 +306,13 @@ async def get_model_profile_image(id: str, user=Depends(get_verified_user)): header, base64_data = model.meta.profile_image_url.split(",", 1) image_data = base64.b64decode(base64_data) image_buffer = io.BytesIO(image_data) + media_type = header.split(";")[0].lstrip("data:") return StreamingResponse( image_buffer, - media_type="image/png", + media_type=media_type, headers={ - "Content-Disposition": "inline; filename=image.png", + "Content-Disposition": "inline", **cache_headers, }, ) diff --git a/backend/open_webui/routers/users.py b/backend/open_webui/routers/users.py index 994e2f2b14..12d1415059 100644 --- a/backend/open_webui/routers/users.py +++ b/backend/open_webui/routers/users.py @@ -461,11 +461,12 @@ async def get_user_profile_image_by_id(user_id: str, user=Depends(get_verified_u header, base64_data = user.profile_image_url.split(",", 1) image_data = base64.b64decode(base64_data) image_buffer = io.BytesIO(image_data) + media_type = header.split(";")[0].lstrip("data:") return StreamingResponse( image_buffer, - media_type="image/png", - headers={"Content-Disposition": "inline; filename=image.png"}, + media_type=media_type, + headers={"Content-Disposition": "inline"}, ) except Exception as e: pass