Spaces:
Running
Running
Commit
·
5100f6b
1
Parent(s):
f50b29d
Add Prodia as an image generation provider and expand model options in generate_image function
Browse files
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:
|