From b84f35f4e4c431cc49a16e1cc1fe47db45407c31 Mon Sep 17 00:00:00 2001
From: hlohaus <983577+hlohaus@users.noreply.github.com>
Date: Sat, 22 Feb 2025 03:33:12 +0100
Subject: Improve model list in HuggingFace WakeLook and disable count tokens
 for performance Export and import settings in UI Some styling improvments in
 UI Fix curl_cffi updated bugs

---
 g4f/Provider/PollinationsImage.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

(limited to 'g4f/Provider/PollinationsImage.py')

diff --git a/g4f/Provider/PollinationsImage.py b/g4f/Provider/PollinationsImage.py
index c9f302f8..bdbc2035 100644
--- a/g4f/Provider/PollinationsImage.py
+++ b/g4f/Provider/PollinationsImage.py
@@ -7,16 +7,17 @@ from ..typing import AsyncResult, Messages
 from .PollinationsAI import PollinationsAI
 
 class PollinationsImage(PollinationsAI):
-    label = "Pollinations AI (Image)"
     default_model = "flux"
     default_vision_model = None
     default_image_model = default_model
+    image_models = [default_image_model]
 
     @classmethod
     def get_models(cls, **kwargs):
-        if not cls.image_models:
-            cls.image_models = list(dict.fromkeys([*cls.image_models, *cls.extra_image_models]))
-        return cls.image_models
+        if not cls.models:
+            super().get_models(**kwargs)
+            cls.models = cls.image_models
+        return cls.models
 
     @classmethod
     async def create_async_generator(
-- 
cgit v1.2.3