Update app.py

app.py

@@ -12,7 +12,7 @@ API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
 API_TOKEN = os.getenv("HF_READ_TOKEN")
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 timeout = 100
-models_list = ["AbsoluteReality 1.8.1", "Playground 2", "Openjourney 4", "Lyriel 1.6", "Animagine XL 2.0", "Counterfeit 2.5", "Realistic Vision 5.1", "Incursios 1.6", "Anime Detailer XL", "Vector Art XL", "epiCRealism", "PixelArt XL", "NewReality XL", "Anything 5.0", "Disney", "CleanLinearMix", "Redmond SDXL", "Arcane"]
+models_list = ["DALL-E 3 XL", "New Reality XL NSFW", "Juggernaut XL", "SDXL 1.0", "AbsoluteReality 1.8.1", "SSD-1B", "Dreamshaper XL Turbo", "Edge of Realism", "Playground 2", "NSFW Hentai", "Openjourney 4", "Lyriel 1.6", "Animagine XL 2.0", "Counterfeit 2.5", "Realistic Vision 5.1", "Incursios 1.6", "Anime Detailer XL", "Vector Art XL", "epiCRealism", "PixelArt XL", "NewReality XL", "Anything 5.0", "Disney", "CleanLinearMix", "Redmond SDXL", "Arcane"]
 
 # OP's other AI projects: https://ai-hub.rf.gd .
 
@@ -76,10 +76,14 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
 
     prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
     print(f'\033[1mГенерация {key}:\033[0m {prompt}')
-    #if model == 'DALL-E 3 XL':
-    #    API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
+    if model == 'DALL-E 3 XL':
+        API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
     if model == 'Playground 2':
         API_URL = "https://api-inference.huggingface.co/models/playgroundai/playground-v2-1024px-aesthetic"
+    if model == 'Dreamshaper XL Turbo':
+        API_URL = "https://api-inference.huggingface.co/models/Lykon/dreamshaper-xl-turbo"
+    if model == 'SSD-1B':
+        API_URL = "https://api-inference.huggingface.co/models/segmind/SSD-1B"
     if model == 'Openjourney 4':
         API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney-v4"
     if model == 'AbsoluteReality 1.8.1':
@@ -117,7 +121,16 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
         API_URL = "https://api-inference.huggingface.co/models/artificialguybr/LogoRedmond-LogoLoraForSDXL-V2"
     if model == 'Arcane':
         API_URL = "https://api-inference.huggingface.co/models/nitrosocke/Arcane-Diffusion"
-
+    if model == 'SDXL 1.0':
+        API_URL = "https://api-inference.huggingface.co/models/stable-diffusion-xl-base-1.0"
+    if model == 'Edge of Realism':
+        API_URL = "https://api-inference.huggingface.co/models/Yntec/edgeOfRealism"
+    if model == 'NSFW Hentai':
+        API_URL = "https://api-inference.huggingface.co/models/explicit-freedom-nsfw-wai"
+    if model == 'New Reality XL NSFW':
+        API_URL = "https://api-inference.huggingface.co/models/newrealityxl-global-nsfw"
+    if model == 'Juggernaut XL':
+        API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/juggernaut-xl-v7"
     
     
     payload = {