summaryrefslogtreecommitdiffstats
path: root/g4f/image.py
diff options
context:
space:
mode:
authorH Lohaus <hlohaus@users.noreply.github.com>2024-05-22 20:18:42 +0200
committerGitHub <noreply@github.com>2024-05-22 20:18:42 +0200
commit6830dfcc6439562af805c8ea72e6e45bba5da401 (patch)
tree30609eb68c513387bd616a14f9cec3b82db6eaff /g4f/image.py
parentImprove readme (diff)
parentFix image vision in You.com provider (diff)
downloadgpt4free-6830dfcc6439562af805c8ea72e6e45bba5da401.tar
gpt4free-6830dfcc6439562af805c8ea72e6e45bba5da401.tar.gz
gpt4free-6830dfcc6439562af805c8ea72e6e45bba5da401.tar.bz2
gpt4free-6830dfcc6439562af805c8ea72e6e45bba5da401.tar.lz
gpt4free-6830dfcc6439562af805c8ea72e6e45bba5da401.tar.xz
gpt4free-6830dfcc6439562af805c8ea72e6e45bba5da401.tar.zst
gpt4free-6830dfcc6439562af805c8ea72e6e45bba5da401.zip
Diffstat (limited to 'g4f/image.py')
-rw-r--r--g4f/image.py7
1 files changed, 7 insertions, 0 deletions
diff --git a/g4f/image.py b/g4f/image.py
index a677e8df..6561b83a 100644
--- a/g4f/image.py
+++ b/g4f/image.py
@@ -16,6 +16,13 @@ from .errors import MissingRequirementsError
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif', 'webp', 'svg'}
+EXTENSIONS_MAP: dict[str, str] = {
+ "image/png": "png",
+ "image/jpeg": "jpg",
+ "image/gif": "gif",
+ "image/webp": "webp",
+}
+
def to_image(image: ImageType, is_svg: bool = False) -> Image:
"""
Converts the input image to a PIL Image object.