Update api.py

2025-10-05 08:16:58 +08:00 · 2024-11-20 10:12:54 +01:00
parent dba41cda56
commit c959d9b469
1 changed files with 1 additions and 1 deletions
--- a/g4f/gui/server/api.py
+++ b/g4f/gui/server/api.py
@@ -48,7 +48,7 @@ class Api:
                        "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
                ]