raannakasturi commited on
Commit
d94ffa5
·
verified ·
1 Parent(s): e1f51f1

Update image.py

Browse files
Files changed (1) hide show
  1. image.py +2 -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, Prodia
9
  from g4f.Provider.hf_space.BlackForestLabsFlux1Schnell import BlackForestLabsFlux1Schnell
10
  from g4f.Provider.hf_space.VoodoohopFlux1Schnell import VoodoohopFlux1Schnell
11
 
@@ -50,8 +50,7 @@ def generate_image(title, category, summary):
50
  ([PollinationsAI], "flux-pro"),
51
  ([Airforce, PollinationsAI, Blackbox], "flux"),
52
  ([PollinationsAI], "dall-e-3"),
53
- ([HuggingSpace], "sd-3"),
54
- ([Prodia], Prodia.default_image_model)
55
  ]
56
 
57
  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
9
  from g4f.Provider.hf_space.BlackForestLabsFlux1Schnell import BlackForestLabsFlux1Schnell
10
  from g4f.Provider.hf_space.VoodoohopFlux1Schnell import VoodoohopFlux1Schnell
11
 
 
50
  ([PollinationsAI], "flux-pro"),
51
  ([Airforce, PollinationsAI, Blackbox], "flux"),
52
  ([PollinationsAI], "dall-e-3"),
53
+ ([HuggingSpace], "sd-3")
 
54
  ]
55
 
56
  for providers, model in attempts: