Update app.py

app.py

@@ -10,7 +10,7 @@ import json
 
 # Project by Nymbo
 
-API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
+API_URL = "https://api-inference.huggingface.co/models/segmind/SSD-1B"
 API_TOKEN = os.getenv("HF_READ_TOKEN")
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 timeout = 100
@@ -82,89 +82,6 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
 #        API_URL = "https://api-inference.huggingface.co/models/"
 #        prompt = f"Ultra realistic porn. {prompt}"
 
-#    if model == 'Xx':
-#        API_URL = "https://api-inference.huggingface.co/models/"
-
-    if model == 'Stable Diffusion XL':
-        API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
-    if model == 'Stable Diffusion 3':
-        API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3-medium-diffusers"
-    if model == 'RealVisXL v4.0':
-        API_URL = "https://api-inference.huggingface.co/models/SG161222/RealVisXL_V4.0"
-    if model == 'Duchaiten Real3D NSFW XL':
-        API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/duchaiten-real3d-nsfw-xl"
-    if model == 'Tempest v0.1':
-        API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/TempestV0.1"
-    if model == 'Explicit Freedom NSFW Waifu':
-        API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/explicit-freedom-nsfw-wai"
-    if model == 'Realistic Stock Photo':
-        API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/realistic-stock-photo"
-    if model == 'Archfey Anime':
-        API_URL = "https://api-inference.huggingface.co/models/alvdansen/archfey_anime"
-    if model == 'Lofi Cuties':
-        API_URL = "https://api-inference.huggingface.co/models/alvdansen/lofi-cuties"
-    if model == 'Mobius':
-        API_URL = "https://api-inference.huggingface.co/models/Corcelio/mobius"
-    if model == 'fennPhoto':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/fennPhoto"
-    if model == 'Juggernaut X v10':
-        API_URL = "https://api-inference.huggingface.co/models/RunDiffusion/Juggernaut-X-v10"
-    if model == 'Chip n DallE':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/Chip_n_DallE"
-    if model == 'Roleplay':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/Roleplay"
-    if model == 'YiffyMix':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/YiffyMix"
-    if model == 'Movie X':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/aMovieX"
-    if model == 'Fabulous':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/Fabulous"
-    if model == 'FotoPhoto':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/FotoPhoto"
-    if model == 'MajicMIX Realistic v6':
-        API_URL = "https://api-inference.huggingface.co/models/digiplay/majicMIX_realistic_v6"
-    if model == 'FX Photorealistic v1':
-        API_URL = "https://api-inference.huggingface.co/models/digiplay/RunDiffusionFXPhotorealistic_v1"
-    if model == 'Analog Madness Realistic v7':
-        API_URL = "https://api-inference.huggingface.co/models/digiplay/AnalogMadness-realistic-model-v7"
-    if model == 'Selfie Photography':
-        API_URL = "https://api-inference.huggingface.co/models/artificialguybr/selfiephotographyredmond-selfie-photography-lora-for-sdxl"
-    if model == 'Filmgrain':
-        API_URL = "https://api-inference.huggingface.co/models/artificialguybr/filmgrain-redmond-filmgrain-lora-for-sdxl"
-    if model == '3D Render Style XL':
-        API_URL = "https://api-inference.huggingface.co/models/goofyai/3d_render_style_xl"
-    if model == 'Leonardo AI Style Illustration':
-        API_URL = "https://api-inference.huggingface.co/models/goofyai/Leonardo_Ai_Style_Illustration"
-    if model == 'Disney Style XL':
-        API_URL = "https://api-inference.huggingface.co/models/goofyai/disney_style_xl"
-    if model == 'Cyborg Style XL':
-        API_URL = "https://api-inference.huggingface.co/models/goofyai/cyborg_style_xl"
-    if model == 'Protogen':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/Protogen_Unofficial_Release"
-    if model == 'Better Pony Diffusiion':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/BetterPonyDiffusion"
-    if model == 'Fluently XL v4':
-        API_URL = "https://api-inference.huggingface.co/models/fluently/Fluently-XL-v4"
-    if model == 'Little Tinies':
-        API_URL = "https://api-inference.huggingface.co/models/alvdansen/littletinies"
-    if model == 'Counterfeit v2.5':
-        API_URL = "https://api-inference.huggingface.co/models/gsdf/Counterfeit-V2.5"
-    if model == 'Raelestia XL':
-        API_URL = "https://api-inference.huggingface.co/models/Raelina/Raelestia-XL"
-    if model == 'Starry XL 5.2':
-        API_URL = "https://api-inference.huggingface.co/models/eienmojiki/Starry-XL-v5.2"
-    if model == 'NSFW Gen 2':
-        API_URL = "https://api-inference.huggingface.co/models/UnfilteredAI/NSFW-gen-v2"
-    if model == 'Mann-E Dreams':
-        API_URL = "https://api-inference.huggingface.co/models/mann-e/Mann-E_Dreams"
-    if model == 'SexOholic':
-        API_URL = "https://api-inference.huggingface.co/models/John6666/sexoholic-real-pony-nsfw-v2-sdxl"
-    if model == 'MIKO Realistic':
-        API_URL = "https://api-inference.huggingface.co/models/John6666/photo-realistic-pony-v5-sdxl"
-    if model == 'Mala Anime Mix':
-        API_URL = "https://api-inference.huggingface.co/models/John6666/photo-realistic-pony-v5-sdxl"
-    if model == 'Lewdfusion':
-        API_URL = "https://api-inference.huggingface.co/models/John6666/getphat-pony-lewdfusion-v1-sdxl"
     if model == 'Fluently XL Final':
         API_URL = "https://api-inference.huggingface.co/models/fluently/Fluently-XL-Final"
     if model == 'NSFW XL':
@@ -181,12 +98,18 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
         API_URL = "https://api-inference.huggingface.co/models/Yntec/Timeless"
     if model == 'Pixel Art Redmond':
         API_URL = "https://api-inference.huggingface.co/models/artificialguybr/PixelArtRedmond"
-    if model == 'Proteus V0.4':
+    if model == 'ProteusV0.4':
         API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/ProteusV0.4"
+    if model == 'ProteusV0.3':
+        API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/ProteusV0.3"
     if model == 'RetroLife':
         API_URL = "https://api-inference.huggingface.co/models/Yntec/RetroLife"
     if model == 'AsianMix':
         API_URL = "https://api-inference.huggingface.co/models/Yntec/AsianMix"
+    if model == 'Stable Diffusion 2.1':
+        API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1"
+    if model == 'Portrait Finetuned':
+        API_URL = "https://api-inference.huggingface.co/models/segmind/portrait-finetuned"
     if model == 'Aurora':
         API_URL = "https://api-inference.huggingface.co/models/Yntec/Aurora"
     if model == 'ShortPrompts':
@@ -268,7 +191,7 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
     if model == 'Incursios 1.6':
         API_URL = "https://api-inference.huggingface.co/models/digiplay/incursiosMemeDiffusion_v1.6"
         prompt = f"Anime porn. {prompt}"
-    if model == 'NewReality XL NSFW':
+    if model == 'NewReality XL':
         API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/newrealityxl-global-nsfw"
     if model == 'Disney':
         API_URL = "https://api-inference.huggingface.co/models/goofyai/disney_style_xl"
@@ -340,57 +263,19 @@ with gr.Blocks(theme='Nymbo/Alyx_Theme') as dalle:
                             
                             # UPDATE WITH NEW MODELS, ORDER MATTERS IN END USER UI
                             
-                            "Stable Diffusion 3",
-                            "Stable Diffusion XL",
-                            "RealVisXL v4.0",
                             "Fluently XL Final",
                             "Animagine XL 3.1",
-                            "Tempest v0.1",
-                            "Juggernaut X v10",
-                            "Mobius",
-                            "fennPhoto",
-                            "Archfey Anime",
-                            "Photon",
-                            "Proteus V0.4",
-                            "Duchaiten Real3D NSFW XL",
-                            "Chip n DallE",
-                            "Roleplay",
-                            "YiffyMix",
-                            "Movie X",
-                            "Fabulous",
-                            "FotoPhoto",
-                            "MajicMIX Realistic v6",
-                            "FX Photorealistic v1",
-                            "Analog Madness Realistic v7",
-                            "Selfie Photography",
-                            "Filmgrain",
-                            "Leonardo AI Style Illustration",
-                            "Disney Style XL",
-                            "Cyborg Style XL",
-                            "Protogen",
-                            "Better Pony Diffusion",
-                            "Fluently XL v4",
-                            "Little Tinies",
-                            "Counterfeit v2.5",
-                            "Starry XL v5.2",
-                            "SexOholic",
-                            "Raelestia XL",
-                            "NSFW Gen 2",
-                            "Mann-E Dreams",
-                            "MIKO Realistic",
-                            "Mala Anime Mix",
-                            "Lewdfusion",
-                            "Explicit Freedom NSFW Waifu",
-                            "Realistic Stock Photo",
-                            "SSD-1B",
                             "Epic Diffusion",
                             "DreamPhotoGasm",
                             "Analog Redmond",
                             "Timeless",
                             "NSFW XL",
                             "Pixel Art Redmond",
+                            "Photon",
                             "Incredible World 2",
                             "ShortPrompts",
+                            "ProteusV0.4",
+                            "ProteusV0.3",
                             "Analog",
                             "RealLife",
                             "Paragon",
@@ -400,8 +285,9 @@ with gr.Blocks(theme='Nymbo/Alyx_Theme') as dalle:
                             "CyberRealistic",
                             "DreamAnything",
                             "pineappleAnimeMix",
-                            "Lofi Cuties",
+                            "Ascii Art",
                             "Aurora",
+                            "Portrait Finetuned",
                             "AsianMix",
                             "OpenGenDiffusers",
                             "NewMoon",
@@ -423,6 +309,7 @@ with gr.Blocks(theme='Nymbo/Alyx_Theme') as dalle:
                             "SDXXXL 2",
                             "epiCPhotoGasm",
                             "AbsoluteReality 1.8.1",
+                            "SSD-1B",
                             "Dreamshaper XL Turbo",
                             "Realistic Vision v12",
                             "NSFW Hentai",
@@ -433,26 +320,22 @@ with gr.Blocks(theme='Nymbo/Alyx_Theme') as dalle:
                             "Incursios 1.6",
                             "SexyToons",
                             "CutesyAnime",
-                            "NewReality XL NSFW",
+                            "NewReality XL",
                             "Disney",
                             "CleanLinearMix",
                             "Redmond SDXL",
                             "SDXL Niji",
                             "Crystal Clear XL",
-                            "Ascii Art",
-                            "Disney",
                             "Playground 2",
+                            "Stable Diffusion 2.1"
                         )
-                        
-                        # SET DEFAULT MODEL HERE
-                        
-                        model = gr.Radio(label="Select a model below", value="Stable Diffusion 3", choices=models_list)
+                        model = gr.Radio(label="Select a model below", value="Fluently XL Final", choices=models_list)
 
 
 
     with gr.Tab("Advanced Settings"):
         with gr.Row():
-            negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="What should not be in the image", value="(deformed, distorted, disfigured), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, misspellings, typos", lines=3, elem_id="negative-prompt-text-input")
+            negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="What should not be in the image", value="[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry, text, fuzziness", lines=3, elem_id="negative-prompt-text-input")
         with gr.Row():
             steps = gr.Slider(label="Sampling steps", value=35, minimum=1, maximum=100, step=1)
         with gr.Row():