Update app.py

app.py

@@ -10,7 +10,7 @@ import json
 
 # Project by Nymbo
 
-API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
+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
@@ -78,11 +78,26 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
     
     # UPDATE WITH MODEL API URL
     
-    if model == 'DALL-E 3 XL':
-        API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
-        prompt = f"Ultra realistic porn. {prompt}"
 #    if model == 'ModelName':
 #        API_URL = "https://api-inference.huggingface.co/models/"
+#        prompt = f"Ultra realistic porn. {prompt}"
+
+    if model == 'Fluently XL Final':
+        API_URL = "https://api-inference.huggingface.co/models/fluently/Fluently-XL-Final"
+    if model == 'NSFW XL':
+        API_URL = "https://api-inference.huggingface.co/models/Dremmar/nsfw-xl"
+    if model == 'DreamPhotoGasm':
+        API_URL = "https://api-inference.huggingface.co/models/Yntec/DreamPhotoGASM"
+    if model == 'Animagine XL 3.1':
+        API_URL = "https://api-inference.huggingface.co/models/cagliostrolab/animagine-xl-3.1"
+    if model == 'Epic Diffusion':
+        API_URL = "https://api-inference.huggingface.co/models/Yntec/EpicDiffusion"
+    if model == 'Analog Redmond':
+        API_URL = "https://api-inference.huggingface.co/models/artificialguybr/analogredmond"
+    if model == 'Timeless':
+        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 == 'ProteusV0.4':
         API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/ProteusV0.4"
     if model == 'ProteusV0.3':
@@ -95,8 +110,6 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
         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 == 'MemeDiffusion':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/MemeDiffusion"
     if model == 'Aurora':
         API_URL = "https://api-inference.huggingface.co/models/Yntec/Aurora"
     if model == 'ShortPrompts':
@@ -137,14 +150,8 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
         API_URL = "https://api-inference.huggingface.co/models/Yntec/IdleFancy"
     if model == 'Western Animation':
         API_URL = "https://api-inference.huggingface.co/models/Yntec/WesternAnimation"
-    if model == 'DreamWorks':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/DreamWorks"
-    if model == 'Mistoon Anime 2':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/mistoonAnime2"
     if model == '3D Animation':
         API_URL = "https://api-inference.huggingface.co/models/Yntec/3Danimation"
-    if model == 'Liberte Redmond':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/LiberteRedmond"
     if model == 'Perfect Level 10':
         API_URL = "https://api-inference.huggingface.co/models/digiplay/perfectlevel10"
     if model == 'Tea':
@@ -181,27 +188,11 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
     if model == 'Animagine XL 2.0':
         API_URL = "https://api-inference.huggingface.co/models/Linaqruf/animagine-xl-2.0"
         prompt = f"Anime porn. {prompt}"
-    if model == 'Counterfeit 2.5':
-        API_URL = "https://api-inference.huggingface.co/models/gsdf/Counterfeit-V2.5"
-    if model == 'Realistic Vision 5.1':
-        API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/realistic-vision-v51"
     if model == 'Incursios 1.6':
         API_URL = "https://api-inference.huggingface.co/models/digiplay/incursiosMemeDiffusion_v1.6"
-    if model == 'Anime Detailer XL':
-        API_URL = "https://api-inference.huggingface.co/models/Linaqruf/anime-detailer-xl-lora"
         prompt = f"Anime porn. {prompt}"
-    if model == 'epiCRealism':
-        API_URL = "https://api-inference.huggingface.co/models/emilianJR/epiCRealism"
-    if model == 'ProteusV0.2':
-        API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/ProteusV0.2"
-    if model == 'PixelArt XL':
-        API_URL = "https://api-inference.huggingface.co/models/nerijs/pixel-art-xl"
     if model == 'NewReality XL':
         API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/newrealityxl-global-nsfw"
-    if model == 'Anything 5.0':
-        API_URL = "https://api-inference.huggingface.co/models/hogiahien/anything-v5-edited"
-    if model == 'Vector Art XL':
-        API_URL = "https://api-inference.huggingface.co/models/DoctorDiffusion/doctor-diffusion-s-controllable-vector-art-xl-lora"
     if model == 'Disney':
         API_URL = "https://api-inference.huggingface.co/models/goofyai/disney_style_xl"
         prompt = f"Disney style. {prompt}"
@@ -209,28 +200,12 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
         API_URL = "https://api-inference.huggingface.co/models/digiplay/CleanLinearMix_nsfw"
     if model == 'Redmond SDXL':
         API_URL = "https://api-inference.huggingface.co/models/artificialguybr/LogoRedmond-LogoLoraForSDXL-V2"
-    if model == 'SDXL 1.0':
-        API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/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/stablediffusionapi/explicit-freedom-nsfw-wai"
-    if model == 'New Reality XL NSFW':
-        API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/newrealityxl-global-nsfw"
-    if model == 'Juggernaut XL':
-        API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/juggernaut-xl-v7"
     if model == 'SDXL Niji':
         API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/SDXL_Niji_SE"
     if model == 'Crystal Clear XL':
         API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/crystal-clear-xlv1"
-    if model == 'NightVision XL':
-        API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/NightVision_XL"
-    if model == 'Elldreth Vivid Mix':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/elldrethSVividMix"
-    if model == 'Deliberate 2':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/Deliberate2"
-    if model == 'Deliberate':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/Deliberate"
     if model == 'SexyToons':
         API_URL = "https://api-inference.huggingface.co/models/Yntec/sexyToons"
     if model == 'Realistic Vision v12':
@@ -276,7 +251,7 @@ css = """
 footer {visibility: hidden !important;}
 """
 
-with gr.Blocks (theme=gr.themes.Default(primary_hue="pink", secondary_hue="pink")) as dalle:
+with gr.Blocks(theme='Nymbo/Alyx_Theme') as dalle:
     with gr.Tab("Basic Settings"):
         with gr.Row():
             with gr.Column(elem_id="prompt-container"):
@@ -287,16 +262,20 @@ with gr.Blocks (theme=gr.themes.Default(primary_hue="pink", secondary_hue="pink"
                         models_list = (
                             
                             # UPDATE WITH NEW MODELS, ORDER MATTERS IN END USER UI
-                            # Currently 77 models
                             
-                            "DALL-E 3 XL",
-                            "SDXL 1.0",
+                            "Fluently XL Final",
+                            "Animagine XL 3.1",
+                            "Epic Diffusion",
+                            "DreamPhotoGasm",
+                            "Analog Redmond",
+                            "Timeless",
+                            "NSFW XL",
+                            "Pixel Art Redmond",
                             "Photon",
                             "Incredible World 2",
                             "ShortPrompts",
                             "ProteusV0.4",
                             "ProteusV0.3",
-                            "ProteusV0.2",
                             "Analog",
                             "RealLife",
                             "Paragon",
@@ -308,7 +287,6 @@ with gr.Blocks (theme=gr.themes.Default(primary_hue="pink", secondary_hue="pink"
                             "pineappleAnimeMix",
                             "Ascii Art",
                             "Aurora",
-                            "MemeDiffusion",
                             "Portrait Finetuned",
                             "AsianMix",
                             "OpenGenDiffusers",
@@ -319,10 +297,7 @@ with gr.Blocks (theme=gr.themes.Default(primary_hue="pink", secondary_hue="pink"
                             "NuipeniMix",
                             "Idle Fancy",
                             "Western Animation",
-                            "DreamWorks",
-                            "Mistoon Anime 2",
                             "3D Animation",
-                            "Liberte Redmond",
                             "Perfect Level 10",
                             "Tea",
                             "AnimeBoysXL 2",
@@ -333,42 +308,28 @@ with gr.Blocks (theme=gr.themes.Default(primary_hue="pink", secondary_hue="pink"
                             "SDXXXL",
                             "SDXXXL 2",
                             "epiCPhotoGasm",
-                            "New Reality XL NSFW",
-                            "Juggernaut XL",
                             "AbsoluteReality 1.8.1",
                             "SSD-1B",
                             "Dreamshaper XL Turbo",
-                            "Edge of Realism",
                             "Realistic Vision v12",
                             "NSFW Hentai",
                             "Lyriel 1.6",
                             "Animagine XL 2.0",
                             "Animagine XL 3.0",
                             "CinemaEros",
-                            "Counterfeit 2.5",
-                            "Deliberate",
-                            "Deliberate 2",
                             "Incursios 1.6",
-                            "Anime Detailer XL",
                             "SexyToons",
                             "CutesyAnime",
-                            "Vector Art XL",
-                            "PixelArt XL",
                             "NewReality XL",
-                            "Anything 5.0",
                             "Disney",
                             "CleanLinearMix",
                             "Redmond SDXL",
-                            "Elldreth Vivid Mix",
                             "SDXL Niji",
                             "Crystal Clear XL",
-                            "NightVision XL",
                             "Playground 2",
-                            "Stable Diffusion 2.1",
-                            "Realistic Vision 5.1",
-                            "epiCRealism"
+                            "Stable Diffusion 2.1"
                         )
-                        model = gr.Radio(label="Select a model below", value="DALL-E 3 XL", choices=models_list)
+                        model = gr.Radio(label="Select a model below", value="Fluently XL Final", choices=models_list)
 
 
 
@@ -388,6 +349,21 @@ with gr.Blocks (theme=gr.themes.Default(primary_hue="pink", secondary_hue="pink"
         # with gr.Row():
         #    gpt = gr.Checkbox(label="ChatGPT")
 
+    with gr.Tab("Image Editor"):
+        def sleep(im):
+            time.sleep(5)
+            return [im["background"], im["layers"][0], im["layers"][1], im["composite"]]
+
+        def predict(im):
+            return im["composite"]
+
+        with gr.Blocks() as demo:
+            with gr.Row():
+                im = gr.ImageEditor(
+                    type="numpy",
+                    crop_size="1:1",
+                )
+
     with gr.Tab("Information"):
         with gr.Row():
             gr.Textbox(label="Sample prompt", value="{prompt} | ultra detail, ultra elaboration, ultra quality, perfect.")