From 4bc4d635bca9c1c7633ff87ff24b757c653ff60f Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Mon, 22 Apr 2024 01:27:48 +0200 Subject: Add vision models to readme --- etc/tool/readme_table.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) (limited to 'etc/tool/readme_table.py') diff --git a/etc/tool/readme_table.py b/etc/tool/readme_table.py index b56e4bca..439b17fa 100644 --- a/etc/tool/readme_table.py +++ b/etc/tool/readme_table.py @@ -127,8 +127,8 @@ def print_models(): def print_image_models(): lines = [ - "| Label | Provider | Model | Website |", - "| ----- | -------- | ----- | ------- |", + "| Label | Provider | Image Model | Vision Model | Website |", + "| ----- | -------- | ----------- | ------------ | ------- |", ] from g4f.gui.server.api import Api for image_model in Api.get_image_models(): @@ -136,13 +136,15 @@ def print_image_models(): netloc = urlparse(provider_url).netloc.replace("www.", "") website = f"[{netloc}]({provider_url})" label = image_model["provider"] if image_model["label"] is None else image_model["label"] - lines.append(f'| {label} | {image_model["provider"]} | {image_model["image_model"]} | {website} |') + if image_model["vision_model"] is None: + image_model["vision_model"] = "❌" + lines.append(f'| {label} | `g4f.Provider.{image_model["provider"]}` | {image_model["image_model"]}| {image_model["vision_model"]} | {website} |') print("\n".join(lines)) if __name__ == "__main__": - print_providers() - print("\n", "-" * 50, "\n") - print_models() + #print_providers() + #print("\n", "-" * 50, "\n") + #print_models() print("\n", "-" * 50, "\n") print_image_models() \ No newline at end of file -- cgit v1.2.3