Update api.py

xtekky · Nov 20, 2024 · c959d9b · c959d9b
1 parent dba41cd
commit c959d9b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/g4f/gui/server/api.py b/g4f/gui/server/api.py
@@ -48,7 +48,7 @@ def get_provider_models(provider: str, api_key: str = None) -> list[dict]:
                         "model": model,
                         "default": model == provider.default_model,
                         "vision": getattr(provider, "default_vision_model", None) == model or model in getattr(provider, "vision_models", []),
-                        "image": model in getattr(provider, "image_models", []),
+                        "image": False if provider.image_models is None else model in provider.image_models,
                     }
                     for model in models
                 ]