From 2d8c2ca2415f0c8f7bbe941d9aa1881359202915 Mon Sep 17 00:00:00 2001
From: kqlio67 <>
Date: Sat, 28 Dec 2024 18:53:47 +0200
Subject: [PATCH] Update g4f/models.py

---
 g4f/models.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/g4f/models.py b/g4f/models.py
index 7a41506d61c..c520f9f824b 100644
--- a/g4f/models.py
+++ b/g4f/models.py
@@ -22,7 +22,6 @@
     HuggingChat,
     HuggingFace,
     HuggingFace,
-    HuggingSpace,
     Liaobots,
     Mhystical,
     Airforce,
@@ -38,6 +37,11 @@
     ReplicateHome,
     RubiksAI,
     TeachAnything,
+    
+    ## HuggingSpace ##
+    BlackForestLabsFlux1Dev,
+    BlackForestLabsFlux1Schnell,
+    VoodoohopFlux1Schnell,
 )
 
 @dataclass(unsafe_hash=True)
@@ -564,13 +568,13 @@ class ImageModel(Model):
 flux_dev = ImageModel(
     name = 'flux-dev',
     base_provider = 'Flux AI',
-    best_provider = IterListProvider([HuggingSpace, HuggingChat, HuggingFace])
+    best_provider = IterListProvider([BlackForestLabsFlux1Dev, HuggingChat, HuggingFace])
 )
 
 flux_schnell = ImageModel(
     name = 'flux-schnell',
     base_provider = 'Flux AI',
-    best_provider = HuggingSpace
+    best_provider = IterListProvider([BlackForestLabsFlux1Schnell, VoodoohopFlux1Schnell])
 )
 
 flux_realism = ImageModel(