Update image.py

image.py

@@ -6,8 +6,8 @@ import time
 from PIL import Image
 from g4f.client import Client
 from g4f.Provider import RetryProvider, PollinationsAI, ImageLabs, Blackbox, HuggingSpace, Airforce
-from g4f.Provider.hf_space.BlackForestLabsFlux1Schnell import BlackForestLabsFlux1Schnell
-from g4f.Provider.hf_space.VoodoohopFlux1Schnell import VoodoohopFlux1Schnell
+from g4f.Provider.hf_space.BlackForestLabs_Flux1Schnell import BlackForestLabs_Flux1Schnell
+from g4f.Provider.hf_space.Voodoohop_Flux1Schnell import Voodoohop_Flux1Schnell
 
 NEGATIVE_PROMPT = (
     "low quality, blurry, pixelated, bad anatomy, bad hands, three hands, three legs, bad arms, missing legs, "
@@ -45,7 +45,7 @@ def generate_image(title, category, summary):
     prompt = f"Generate accurate image representing the {category} concept: ```{title.strip()}: {summary.strip()}```"
     client = Client()
     attempts = [
-        ([BlackForestLabsFlux1Schnell, VoodoohopFlux1Schnell, HuggingSpace], "flux-schnell"),
+        ([BlackForestLabs_Flux1Schnell, Voodoohop_Flux1Schnell, HuggingSpace], "flux-schnell"),
         ([HuggingSpace, PollinationsAI], "flux-dev"),
         ([PollinationsAI], "flux-pro"),
         ([Airforce, PollinationsAI, Blackbox], "flux"),
@@ -117,4 +117,4 @@ if __name__ == "__main__":
         "It details the calculation of these invariants using moduli spaces of stable maps and explores their implications in enumerative geometry."
     )
     image_url = fetch_image(title, category, summary)
-    print(image_url)
+    print(image_url)