Fix missing model in unittest

hlohaus · Jan 26, 2025 · ca323b1 · ca323b1
1 parent ab04d1d
commit ca323b1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/g4f/Provider/needs_auth/HuggingFaceAPI.py b/g4f/Provider/needs_auth/HuggingFaceAPI.py
@@ -19,9 +19,9 @@ class HuggingFaceAPI(OpenaiTemplate):
 
     @classmethod
     def get_models(cls, **kwargs):
-        HuggingChat.get_models()
-        cls.models = HuggingChat.text_models
-        cls.vision_models = HuggingChat.vision_models
+        if not cls.models:
+            HuggingChat.get_models()
+            cls.models = list(set(HuggingChat.text_models + cls.vision_models))
         return cls.models
 
     @classmethod