Update image_engine.py

image_engine.py

@@ -1,13 +1,17 @@
 from PIL import Image
 import requests
+import random
 import io
 import os
 
 # Define the ImageGenerator class for Hugging Face API
 class ImageGenerator:
     def __init__(self) -> None:
-        # self.api_url = "https://api-inference.huggingface.co/models/stablediffusionapi/mklan-xxx-nsfw-pony" 
-        self.api_url = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large"
+        self.all_urls = [
+            "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell",
+            "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large"
+        ]
+        self.api_url = random.choice(self.all_urls)
         self.headers = {"Authorization": f"Bearer {os.getenv('HUGGING_FACE')}"}
     
     def generate_image(self, prompt):