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

Add Prodia as an image generation provider and expand model options in generate_image function

Browse files
Files changed (1) hide show
  1. image.py +7 -3
image.py CHANGED
@@ -5,7 +5,7 @@ import re
5
  import time
6
  from PIL import Image
7
  from g4f.client import Client
8
- from g4f.Provider import RetryProvider, PollinationsAI, ImageLabs, Blackbox, HuggingSpace, Airforce
9
  from g4f.Provider.hf_space.BlackForestLabsFlux1Schnell import BlackForestLabsFlux1Schnell
10
  from g4f.Provider.hf_space.VoodoohopFlux1Schnell import VoodoohopFlux1Schnell
11
 
@@ -45,9 +45,13 @@ 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
- ([ImageLabs, PollinationsAI], "sdxl-turbo"),
49
  ([Airforce, PollinationsAI, Blackbox], "flux"),
50
- ([BlackForestLabsFlux1Schnell, VoodoohopFlux1Schnell, HuggingSpace], "flux-schnell")
 
 
 
 
 
51
  ]
52
 
53
  for providers, model in attempts:
 
5
  import time
6
  from PIL import Image
7
  from g4f.client import Client
8
+ from g4f.Provider import RetryProvider, PollinationsAI, ImageLabs, Blackbox, HuggingSpace, Airforce, Prodia
9
  from g4f.Provider.hf_space.BlackForestLabsFlux1Schnell import BlackForestLabsFlux1Schnell
10
  from g4f.Provider.hf_space.VoodoohopFlux1Schnell import VoodoohopFlux1Schnell
11
 
 
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)
55
  ]
56
 
57
  for providers, model in attempts: