raannakasturi commited on
Commit
e1f51f1
·
1 Parent(s): 5100f6b

Reorder image generation providers to include Airforce, PollinationsAI, and Blackbox for the flux model

Browse files
Files changed (1) hide show
  1. image.py +1 -1
image.py CHANGED
@@ -45,10 +45,10 @@ def generate_image(title, category, summary):
45
  prompt = f"Generate accurate image representing the {category} concept: ```{title.strip()}: {summary.strip()}```"
46
  client = Client()
47
  attempts = [
48
- ([Airforce, PollinationsAI, Blackbox], "flux"),
49
  ([BlackForestLabsFlux1Schnell, VoodoohopFlux1Schnell, HuggingSpace], "flux-schnell"),
50
  ([HuggingSpace, PollinationsAI], "flux-dev"),
51
  ([PollinationsAI], "flux-pro"),
 
52
  ([PollinationsAI], "dall-e-3"),
53
  ([HuggingSpace], "sd-3"),
54
  ([Prodia], Prodia.default_image_model)
 
45
  prompt = f"Generate accurate image representing the {category} concept: ```{title.strip()}: {summary.strip()}```"
46
  client = Client()
47
  attempts = [
 
48
  ([BlackForestLabsFlux1Schnell, VoodoohopFlux1Schnell, HuggingSpace], "flux-schnell"),
49
  ([HuggingSpace, PollinationsAI], "flux-dev"),
50
  ([PollinationsAI], "flux-pro"),
51
+ ([Airforce, PollinationsAI, Blackbox], "flux"),
52
  ([PollinationsAI], "dall-e-3"),
53
  ([HuggingSpace], "sd-3"),
54
  ([Prodia], Prodia.default_image_model)