Adding support for custom inference models/LoRAs

app.py

@@ -10,19 +10,20 @@ import json
 # Project by Nymbo
 
 # Base API URL for Hugging Face inference
-API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
+API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
 # Retrieve the API token from environment variables
 API_TOKEN = os.getenv("HF_READ_TOKEN")
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 # Timeout for requests
 timeout = 100
 
-def query(prompt, model, is_negative=False, steps=35, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
+def query(prompt, model, custom_lora, is_negative=False, steps=35, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
     # Debug log to indicate function start
     print("Starting query function...")
     # Print the parameters for debugging purposes
     print(f"Prompt: {prompt}")
     print(f"Model: {model}")
+    print(f"Custom LoRA: {custom_lora}")
     print(f"Parameters - Steps: {steps}, CFG Scale: {cfg_scale}, Seed: {seed}, Strength: {strength}, Width: {width}, Height: {height}")
 
     # Check if the prompt is empty or None
@@ -43,211 +44,209 @@ def query(prompt, model, is_negative=False, steps=35, cfg_scale=7, sampler="DPM+
     prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
     print(f'Generation {key}: {prompt}')  # Debug log
 
-    # Set the API URL based on the selected model
-    
-#    if model == 'X':
-#        API_URL = "https://api-inference.huggingface.co/models/X"
-#        prompt = f"X, {prompt}"
-
-    if model == 'Stable Diffusion XL':
-        API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
-    if model == 'FLUX.1 [Dev]':
-        API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
-    if model == 'FLUX.1 [Schnell]':
-        API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
-    if model == 'Flux Logo Design':
-        API_URL = "https://api-inference.huggingface.co/models/Shakker-Labs/FLUX.1-dev-LoRA-Logo-Design"
-        prompt = f"wablogo, logo, Minimalist, {prompt}"
-    if model == 'Flux Uncensored':
-        API_URL = "https://api-inference.huggingface.co/models/enhanceaiteam/Flux-uncensored"
-    if model == 'Flux Uncensored V2':
-        API_URL = "https://api-inference.huggingface.co/models/enhanceaiteam/Flux-Uncensored-V2"
-    if model == 'Flux Tarot Cards':
-        API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Ton618-Tarot-Cards-Flux-LoRA"
-        prompt = f"Tarot card, {prompt}"
-    if model == 'Pixel Art Sprites':
-        API_URL = "https://api-inference.huggingface.co/models/sWizad/pokemon-trainer-sprites-pixelart-flux"
-        prompt = f"a pixel image, {prompt}"
-    if model == '3D Sketchfab':
-        API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Castor-3D-Sketchfab-Flux-LoRA"
-        prompt = f"3D Sketchfab, {prompt}"
-    if model == 'Retro Comic Flux':
-        API_URL = "https://api-inference.huggingface.co/models/renderartist/retrocomicflux"
-        prompt = f"c0m1c, comic book panel, {prompt}"
-    if model == 'Caricature':
-        API_URL = "https://api-inference.huggingface.co/models/TheAwakenOne/caricature"
-        prompt = f"CCTUR3, {prompt}"
-    if model == 'Huggieverse':
-        API_URL = "https://api-inference.huggingface.co/models/Chunte/flux-lora-Huggieverse"
-        prompt = f"HGGRE, {prompt}"
-    if model == 'Propaganda Poster':
-        API_URL = "https://api-inference.huggingface.co/models/AlekseyCalvin/Propaganda_Poster_Schnell_by_doctor_diffusion"
-        prompt = f"propaganda poster, {prompt}"
-    if model == 'Flux Game Assets V2':
-        API_URL = "https://api-inference.huggingface.co/models/gokaygokay/Flux-Game-Assets-LoRA-v2"
-        prompt = f"wbgmsst, white background, {prompt}"
-    if model == 'SoftPasty Flux':
-        API_URL = "https://api-inference.huggingface.co/models/alvdansen/softpasty-flux-dev"
-        prompt = f"araminta_illus illustration style, {prompt}"
-    if model == 'Flux Stickers':
-        API_URL = "https://api-inference.huggingface.co/models/diabolic6045/Flux_Sticker_Lora"
-        prompt = f"5t1cker 5ty1e, {prompt}"
-    if model == 'Flux Animex V2':
-        API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Animex-v2-LoRA"
-        prompt = f"Animex, {prompt}"
-    if model == 'Flux Animeo V1':
-        API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Animeo-v1-LoRA"
-        prompt = f"Animeo, {prompt}"
-    if model == 'Movie Board':
-        API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Flux.1-Dev-Movie-Boards-LoRA"
-        prompt = f"movieboard, {prompt}"
-    if model == 'Purple Dreamy':
-        API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Purple-Dreamy-Flux-LoRA"
-        prompt = f"Purple Dreamy, {prompt}"
-    if model == 'PS1 Style Flux':
-        API_URL = "https://api-inference.huggingface.co/models/veryVANYA/ps1-style-flux"
-        prompt = f"ps1 game screenshot, {prompt}"
-    if model == 'Softserve Anime':
-        API_URL = "https://api-inference.huggingface.co/models/alvdansen/softserve_anime"
-        prompt = f"sftsrv style illustration, {prompt}"
-    if model == 'Flux Tarot v1':
-        API_URL = "https://api-inference.huggingface.co/models/multimodalart/flux-tarot-v1"
-        prompt = f"in the style of TOK a trtcrd tarot style, {prompt}"
-    if model == 'Half Illustration':
-        API_URL = "https://api-inference.huggingface.co/models/davisbro/half_illustration"
-        prompt = f"in the style of TOK, {prompt}"
-    if model == 'OpenDalle v1.1':
-        API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/OpenDalleV1.1"
-    if model == 'Flux Ghibsky Illustration':
-        API_URL = "https://api-inference.huggingface.co/models/aleksa-codes/flux-ghibsky-illustration"
-        prompt = f"GHIBSKY style, {prompt}"
-    if model == 'Flux Koda':
-        API_URL = "https://api-inference.huggingface.co/models/alvdansen/flux-koda"
-        prompt = f"flmft style, {prompt}"
-    if model == 'Soviet Diffusion XL':
-        API_URL = "https://api-inference.huggingface.co/models/openskyml/soviet-diffusion-xl"
-        prompt = f"soviet poster, {prompt}"
-    if model == 'Flux Realism LoRA':
-        API_URL = "https://api-inference.huggingface.co/models/XLabs-AI/flux-RealismLora"
-    if model == 'Frosting Lane Flux':
-        API_URL = "https://api-inference.huggingface.co/models/alvdansen/frosting_lane_flux"
-        prompt = f"frstingln illustration, {prompt}"
-    if model == 'Phantasma Anime':
-        API_URL = "https://api-inference.huggingface.co/models/alvdansen/phantasma-anime"
-    if model == 'Boreal':
-        API_URL = "https://api-inference.huggingface.co/models/kudzueye/Boreal"
-        prompt = f"photo, {prompt}"
-    if model == 'How2Draw':
-        API_URL = "https://api-inference.huggingface.co/models/glif/how2draw"
-        prompt = f"How2Draw, {prompt}"
-    if model == 'Flux AestheticAnime':
-        API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/FLUX-AestheticAnime"
-    if model == 'Fashion Hut Modeling LoRA':
-        API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Fashion-Hut-Modeling-LoRA"
-        prompt = f"Modeling of, {prompt}"
-    if model == 'Flux SyntheticAnime':
-        API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/FLUX-SyntheticAnime"
-        prompt = f"1980s anime screengrab, VHS quality, syntheticanime, {prompt}"
-    if model == 'Flux Midjourney Anime':
-        API_URL = "https://api-inference.huggingface.co/models/brushpenbob/flux-midjourney-anime"
-        prompt = f"egmid, {prompt}"
-    if model == 'Coloring Book Generator':
-        API_URL = "https://api-inference.huggingface.co/models/robert123231/coloringbookgenerator"
-    if model == 'Collage Flux':
-        API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Castor-Collage-Dim-Flux-LoRA"
-        prompt = f"collage, {prompt}"
-    if model == 'Flux Product Ad Backdrop':
-        API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Flux-Product-Ad-Backdrop"
-        prompt = f"Product Ad, {prompt}"
-    if model == 'Product Design':
-        API_URL = "https://api-inference.huggingface.co/models/multimodalart/product-design"
-        prompt = f"product designed by prdsgn, {prompt}"
-    if model == '90s Anime Art':
-        API_URL = "https://api-inference.huggingface.co/models/glif/90s-anime-art"
-    if model == 'Brain Melt Acid Art':
-        API_URL = "https://api-inference.huggingface.co/models/glif/Brain-Melt-Acid-Art"
-        prompt = f"maximalism, in an acid surrealism style, {prompt}"
-    if model == 'Lustly Flux Uncensored v1':
-        API_URL = "https://api-inference.huggingface.co/models/lustlyai/Flux_Lustly.ai_Uncensored_nsfw_v1"
-    if model == 'NSFW Master Flux':
-        API_URL = "https://api-inference.huggingface.co/models/Keltezaa/NSFW_MASTER_FLUX"
-        prompt = f"NSFW, {prompt}"
-    if model == 'Flux Outfit Generator':
-        API_URL = "https://api-inference.huggingface.co/models/tryonlabs/FLUX.1-dev-LoRA-Outfit-Generator"
-    if model == 'Midjourney':
-        API_URL = "https://api-inference.huggingface.co/models/Jovie/Midjourney"
-    if model == 'DreamPhotoGASM':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/DreamPhotoGASM"
-    if model == 'Flux Super Realism LoRA':
-        API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Super-Realism-LoRA"
-    if model == 'Stable Diffusion 2-1':
-        API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1-base"
-    if model == 'Stable Diffusion 3.5 Large':
-        API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large"
-    if model == 'Stable Diffusion 3.5 Large Turbo':
-        API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large-turbo"
-    if model == 'Stable Diffusion 3 Medium':
-        API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3-medium-diffusers"
-        prompt = f"A, {prompt}"
-    if model == 'Duchaiten Real3D NSFW XL':
-        API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/duchaiten-real3d-nsfw-xl"
-    if model == 'Pixel Art XL':
-        API_URL = "https://api-inference.huggingface.co/models/nerijs/pixel-art-xl"
-        prompt = f"pixel art, {prompt}"
-    if model == 'Character Design':
-        API_URL = "https://api-inference.huggingface.co/models/KappaNeuro/character-design"
-        prompt = f"Character Design, {prompt}"
-    if model == 'Sketched Out Manga':
-        API_URL = "https://api-inference.huggingface.co/models/alvdansen/sketchedoutmanga"
-        prompt = f"daiton, {prompt}"
-    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 == 'YiffyMix':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/YiffyMix"
-    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"
-        prompt = f"instagram model, discord profile picture, {prompt}"
-    if model == 'Filmgrain':
-        API_URL = "https://api-inference.huggingface.co/models/artificialguybr/filmgrain-redmond-filmgrain-lora-for-sdxl"
-        prompt = f"Film Grain, FilmGrainAF, {prompt}"
-    if model == 'Leonardo AI Style Illustration':
-        API_URL = "https://api-inference.huggingface.co/models/goofyai/Leonardo_Ai_Style_Illustration"
-        prompt = f"leonardo style, illustration, vector art, {prompt}"
-    if model == 'Cyborg Style XL':
-        API_URL = "https://api-inference.huggingface.co/models/goofyai/cyborg_style_xl"
-        prompt = f"cyborg style, {prompt}"
-    if model == 'Little Tinies':
-        API_URL = "https://api-inference.huggingface.co/models/alvdansen/littletinies"
-    if model == 'NSFW XL':
-        API_URL = "https://api-inference.huggingface.co/models/Dremmar/nsfw-xl"
-    if model == 'Analog Redmond':
-        API_URL = "https://api-inference.huggingface.co/models/artificialguybr/analogredmond"
-        prompt = f"timeless style, {prompt}"
-    if model == 'Pixel Art Redmond':
-        API_URL = "https://api-inference.huggingface.co/models/artificialguybr/PixelArtRedmond"
-        prompt = f"Pixel Art, {prompt}"
-    if model == 'Ascii Art':
-        API_URL = "https://api-inference.huggingface.co/models/CiroN2022/ascii-art"
-        prompt = f"ascii art, {prompt}"
-    if model == 'Analog':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/Analog"
-    if model == 'Maple Syrup':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/MapleSyrup"
-    if model == 'Perfect Lewd Fantasy':
-        API_URL = "https://api-inference.huggingface.co/models/digiplay/perfectLewdFantasy_v1.01"
-    if model == 'AbsoluteReality 1.8.1':
-        API_URL = "https://api-inference.huggingface.co/models/digiplay/AbsoluteReality_v1.8.1"
-    if model == 'Disney':
-        API_URL = "https://api-inference.huggingface.co/models/goofyai/disney_style_xl"
-        prompt = f"Disney style, {prompt}"
-    if model == 'Redmond SDXL':
-        API_URL = "https://api-inference.huggingface.co/models/artificialguybr/LogoRedmond-LogoLoraForSDXL-V2"
-    if model == 'epiCPhotoGasm':
-        API_URL = "https://api-inference.huggingface.co/models/Yntec/epiCPhotoGasm"
+    # Set the API URL based on the selected model or custom LoRA
+    if custom_lora.strip() != "":
+        API_URL = f"https://api-inference.huggingface.co/models/{custom_lora.strip()}"
+    else:
+        if model == 'Stable Diffusion XL':
+            API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
+        if model == 'FLUX.1 [Dev]':
+            API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
+        if model == 'FLUX.1 [Schnell]':
+            API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
+        if model == 'Flux Logo Design':
+            API_URL = "https://api-inference.huggingface.co/models/Shakker-Labs/FLUX.1-dev-LoRA-Logo-Design"
+            prompt = f"wablogo, logo, Minimalist, {prompt}"
+        if model == 'Flux Uncensored':
+            API_URL = "https://api-inference.huggingface.co/models/enhanceaiteam/Flux-uncensored"
+        if model == 'Flux Uncensored V2':
+            API_URL = "https://api-inference.huggingface.co/models/enhanceaiteam/Flux-Uncensored-V2"
+        if model == 'Flux Tarot Cards':
+            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Ton618-Tarot-Cards-Flux-LoRA"
+            prompt = f"Tarot card, {prompt}"
+        if model == 'Pixel Art Sprites':
+            API_URL = "https://api-inference.huggingface.co/models/sWizad/pokemon-trainer-sprites-pixelart-flux"
+            prompt = f"a pixel image, {prompt}"
+        if model == '3D Sketchfab':
+            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Castor-3D-Sketchfab-Flux-LoRA"
+            prompt = f"3D Sketchfab, {prompt}"
+        if model == 'Retro Comic Flux':
+            API_URL = "https://api-inference.huggingface.co/models/renderartist/retrocomicflux"
+            prompt = f"c0m1c, comic book panel, {prompt}"
+        if model == 'Caricature':
+            API_URL = "https://api-inference.huggingface.co/models/TheAwakenOne/caricature"
+            prompt = f"CCTUR3, {prompt}"
+        if model == 'Huggieverse':
+            API_URL = "https://api-inference.huggingface.co/models/Chunte/flux-lora-Huggieverse"
+            prompt = f"HGGRE, {prompt}"
+        if model == 'Propaganda Poster':
+            API_URL = "https://api-inference.huggingface.co/models/AlekseyCalvin/Propaganda_Poster_Schnell_by_doctor_diffusion"
+            prompt = f"propaganda poster, {prompt}"
+        if model == 'Flux Game Assets V2':
+            API_URL = "https://api-inference.huggingface.co/models/gokaygokay/Flux-Game-Assets-LoRA-v2"
+            prompt = f"wbgmsst, white background, {prompt}"
+        if model == 'SoftPasty Flux':
+            API_URL = "https://api-inference.huggingface.co/models/alvdansen/softpasty-flux-dev"
+            prompt = f"araminta_illus illustration style, {prompt}"
+        if model == 'Flux Stickers':
+            API_URL = "https://api-inference.huggingface.co/models/diabolic6045/Flux_Sticker_Lora"
+            prompt = f"5t1cker 5ty1e, {prompt}"
+        if model == 'Flux Animex V2':
+            API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Animex-v2-LoRA"
+            prompt = f"Animex, {prompt}"
+        if model == 'Flux Animeo V1':
+            API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Animeo-v1-LoRA"
+            prompt = f"Animeo, {prompt}"
+        if model == 'Movie Board':
+            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Flux.1-Dev-Movie-Boards-LoRA"
+            prompt = f"movieboard, {prompt}"
+        if model == 'Purple Dreamy':
+            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Purple-Dreamy-Flux-LoRA"
+            prompt = f"Purple Dreamy, {prompt}"
+        if model == 'PS1 Style Flux':
+            API_URL = "https://api-inference.huggingface.co/models/veryVANYA/ps1-style-flux"
+            prompt = f"ps1 game screenshot, {prompt}"
+        if model == 'Softserve Anime':
+            API_URL = "https://api-inference.huggingface.co/models/alvdansen/softserve_anime"
+            prompt = f"sftsrv style illustration, {prompt}"
+        if model == 'Flux Tarot v1':
+            API_URL = "https://api-inference.huggingface.co/models/multimodalart/flux-tarot-v1"
+            prompt = f"in the style of TOK a trtcrd tarot style, {prompt}"
+        if model == 'Half Illustration':
+            API_URL = "https://api-inference.huggingface.co/models/davisbro/half_illustration"
+            prompt = f"in the style of TOK, {prompt}"
+        if model == 'OpenDalle v1.1':
+            API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/OpenDalleV1.1"
+        if model == 'Flux Ghibsky Illustration':
+            API_URL = "https://api-inference.huggingface.co/models/aleksa-codes/flux-ghibsky-illustration"
+            prompt = f"GHIBSKY style, {prompt}"
+        if model == 'Flux Koda':
+            API_URL = "https://api-inference.huggingface.co/models/alvdansen/flux-koda"
+            prompt = f"flmft style, {prompt}"
+        if model == 'Soviet Diffusion XL':
+            API_URL = "https://api-inference.huggingface.co/models/openskyml/soviet-diffusion-xl"
+            prompt = f"soviet poster, {prompt}"
+        if model == 'Flux Realism LoRA':
+            API_URL = "https://api-inference.huggingface.co/models/XLabs-AI/flux-RealismLora"
+        if model == 'Frosting Lane Flux':
+            API_URL = "https://api-inference.huggingface.co/models/alvdansen/frosting_lane_flux"
+            prompt = f"frstingln illustration, {prompt}"
+        if model == 'Phantasma Anime':
+            API_URL = "https://api-inference.huggingface.co/models/alvdansen/phantasma-anime"
+        if model == 'Boreal':
+            API_URL = "https://api-inference.huggingface.co/models/kudzueye/Boreal"
+            prompt = f"photo, {prompt}"
+        if model == 'How2Draw':
+            API_URL = "https://api-inference.huggingface.co/models/glif/how2draw"
+            prompt = f"How2Draw, {prompt}"
+        if model == 'Flux AestheticAnime':
+            API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/FLUX-AestheticAnime"
+        if model == 'Fashion Hut Modeling LoRA':
+            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Fashion-Hut-Modeling-LoRA"
+            prompt = f"Modeling of, {prompt}"
+        if model == 'Flux SyntheticAnime':
+            API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/FLUX-SyntheticAnime"
+            prompt = f"1980s anime screengrab, VHS quality, syntheticanime, {prompt}"
+        if model == 'Flux Midjourney Anime':
+            API_URL = "https://api-inference.huggingface.co/models/brushpenbob/flux-midjourney-anime"
+            prompt = f"egmid, {prompt}"
+        if model == 'Coloring Book Generator':
+            API_URL = "https://api-inference.huggingface.co/models/robert123231/coloringbookgenerator"
+        if model == 'Collage Flux':
+            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Castor-Collage-Dim-Flux-LoRA"
+            prompt = f"collage, {prompt}"
+        if model == 'Flux Product Ad Backdrop':
+            API_URL = "https://api-inference.huggingface.co/models/prithivMLmods/Flux-Product-Ad-Backdrop"
+            prompt = f"Product Ad, {prompt}"
+        if model == 'Product Design':
+            API_URL = "https://api-inference.huggingface.co/models/multimodalart/product-design"
+            prompt = f"product designed by prdsgn, {prompt}"
+        if model == '90s Anime Art':
+            API_URL = "https://api-inference.huggingface.co/models/glif/90s-anime-art"
+        if model == 'Brain Melt Acid Art':
+            API_URL = "https://api-inference.huggingface.co/models/glif/Brain-Melt-Acid-Art"
+            prompt = f"maximalism, in an acid surrealism style, {prompt}"
+        if model == 'Lustly Flux Uncensored v1':
+            API_URL = "https://api-inference.huggingface.co/models/lustlyai/Flux_Lustly.ai_Uncensored_nsfw_v1"
+        if model == 'NSFW Master Flux':
+            API_URL = "https://api-inference.huggingface.co/models/Keltezaa/NSFW_MASTER_FLUX"
+            prompt = f"NSFW, {prompt}"
+        if model == 'Flux Outfit Generator':
+            API_URL = "https://api-inference.huggingface.co/models/tryonlabs/FLUX.1-dev-LoRA-Outfit-Generator"
+        if model == 'Midjourney':
+            API_URL = "https://api-inference.huggingface.co/models/Jovie/Midjourney"
+        if model == 'DreamPhotoGASM':
+            API_URL = "https://api-inference.huggingface.co/models/Yntec/DreamPhotoGASM"
+        if model == 'Flux Super Realism LoRA':
+            API_URL = "https://api-inference.huggingface.co/models/strangerzonehf/Flux-Super-Realism-LoRA"
+        if model == 'Stable Diffusion 2-1':
+            API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1-base"
+        if model == 'Stable Diffusion 3.5 Large':
+            API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large"
+        if model == 'Stable Diffusion 3.5 Large Turbo':
+            API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large-turbo"
+        if model == 'Stable Diffusion 3 Medium':
+            API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3-medium-diffusers"
+            prompt = f"A, {prompt}"
+        if model == 'Duchaiten Real3D NSFW XL':
+            API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/duchaiten-real3d-nsfw-xl"
+        if model == 'Pixel Art XL':
+            API_URL = "https://api-inference.huggingface.co/models/nerijs/pixel-art-xl"
+            prompt = f"pixel art, {prompt}"
+        if model == 'Character Design':
+            API_URL = "https://api-inference.huggingface.co/models/KappaNeuro/character-design"
+            prompt = f"Character Design, {prompt}"
+        if model == 'Sketched Out Manga':
+            API_URL = "https://api-inference.huggingface.co/models/alvdansen/sketchedoutmanga"
+            prompt = f"daiton, {prompt}"
+        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 == 'YiffyMix':
+            API_URL = "https://api-inference.huggingface.co/models/Yntec/YiffyMix"
+        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"
+            prompt = f"instagram model, discord profile picture, {prompt}"
+        if model == 'Filmgrain':
+            API_URL = "https://api-inference.huggingface.co/models/artificialguybr/filmgrain-redmond-filmgrain-lora-for-sdxl"
+            prompt = f"Film Grain, FilmGrainAF, {prompt}"
+        if model == 'Leonardo AI Style Illustration':
+            API_URL = "https://api-inference.huggingface.co/models/goofyai/Leonardo_Ai_Style_Illustration"
+            prompt = f"leonardo style, illustration, vector art, {prompt}"
+        if model == 'Cyborg Style XL':
+            API_URL = "https://api-inference.huggingface.co/models/goofyai/cyborg_style_xl"
+            prompt = f"cyborg style, {prompt}"
+        if model == 'Little Tinies':
+            API_URL = "https://api-inference.huggingface.co/models/alvdansen/littletinies"
+        if model == 'NSFW XL':
+            API_URL = "https://api-inference.huggingface.co/models/Dremmar/nsfw-xl"
+        if model == 'Analog Redmond':
+            API_URL = "https://api-inference.huggingface.co/models/artificialguybr/analogredmond"
+            prompt = f"timeless style, {prompt}"
+        if model == 'Pixel Art Redmond':
+            API_URL = "https://api-inference.huggingface.co/models/artificialguybr/PixelArtRedmond"
+            prompt = f"Pixel Art, {prompt}"
+        if model == 'Ascii Art':
+            API_URL = "https://api-inference.huggingface.co/models/CiroN2022/ascii-art"
+            prompt = f"ascii art, {prompt}"
+        if model == 'Analog':
+            API_URL = "https://api-inference.huggingface.co/models/Yntec/Analog"
+        if model == 'Maple Syrup':
+            API_URL = "https://api-inference.huggingface.co/models/Yntec/MapleSyrup"
+        if model == 'Perfect Lewd Fantasy':
+            API_URL = "https://api-inference.huggingface.co/models/digiplay/perfectLewdFantasy_v1.01"
+        if model == 'AbsoluteReality 1.8.1':
+            API_URL = "https://api-inference.huggingface.co/models/digiplay/AbsoluteReality_v1.8.1"
+        if model == 'Disney':
+            API_URL = "https://api-inference.huggingface.co/models/goofyai/disney_style_xl"
+            prompt = f"Disney style, {prompt}"
+        if model == 'Redmond SDXL':
+            API_URL = "https://api-inference.huggingface.co/models/artificialguybr/LogoRedmond-LogoLoraForSDXL-V2"
+        if model == 'epiCPhotoGasm':
+            API_URL = "https://api-inference.huggingface.co/models/Yntec/epiCPhotoGasm"
     print(f"API URL set to: {API_URL}")  # Debug log
 
     # Define the payload for the request
@@ -290,7 +289,7 @@ def query(prompt, model, is_negative=False, steps=35, cfg_scale=7, sampler="DPM+
             raise gr.Error(f"{response.status_code}: The model is being loaded. Please try again later.")
         else:
             raise gr.Error(f"{response.status_code}: An unexpected error occurred.")
-    
+
     try:
         # Attempt to read the image from the response content
         image_bytes = response.content
@@ -319,11 +318,14 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme_5') as dalle:
                 with gr.Row():
                     # Textbox for user to input the prompt
                     text_prompt = gr.Textbox(label="Prompt", placeholder="Enter a prompt here", lines=3, elem_id="prompt-text-input")
+                with gr.Row():
+                    # Textbox for custom LoRA input
+                    custom_lora = gr.Textbox(label="Custom LoRA", info="LoRA Hugging Face path (optional)", placeholder="multimodalart/vintage-ads-flux")
                 with gr.Row():
                     # Accordion for selecting the model
-                    with gr.Accordion("Model Selection", open=True):
+                    with gr.Accordion("Featured Models", open=True):
                         # Textbox for searching models
-                        model_search = gr.Textbox(label="Search Models", placeholder="Search for a model...", lines=1, elem_id="model-search-input")
+                        model_search = gr.Textbox(label="Filter Models", placeholder="Search for a featured model...", lines=1, elem_id="model-search-input")
                         models_list = (
                             "3D Sketchfab",
                             "90s Anime Art",
@@ -467,11 +469,12 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme_5') as dalle:
         with gr.Row():
             # Display a sample prompt for guidance
             gr.Textbox(label="Sample prompt", value="{prompt} | ultra detail, ultra elaboration, ultra quality, perfect.")
-            
+
         # Accordion displaying featured models
         with gr.Accordion("Featured Models (WiP)", open=False):
             gr.HTML(
                 """
+            <p><a href="https://huggingface.co/models?inference=warm&sort=likes&search=stable-diffusion">See all available models</a></p>
             <table style="width:100%; text-align:center; margin:auto;">
                 <tr>
                     <th>Model Name</th>
@@ -494,14 +497,13 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme_5') as dalle:
                     <td></td>
                 </tr>
             </table>
-
             """
             )
 
         # Accordion providing an overview of advanced settings
         with gr.Accordion("Advanced Settings Overview", open=False):
             gr.Markdown(
-                """
+            """
             ## Negative Prompt
             ###### This box is for telling the AI what you don't want in your images. Think of it as a way to avoid certain elements. For instance, if you don't want blurry images or extra limbs showing up, this is where you'd mention it.
 
@@ -513,7 +515,7 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme_5') as dalle:
 
             ## CFG Scale
             ###### CFG stands for "Control Free Guidance." The scale adjusts how closely the AI follows your prompt. A lower number makes the AI more creative and free-flowing, while a higher number makes it stick closely to what you asked for. If you want the AI to take fewer artistic liberties, slide this towards a higher number. Just think "Control Freak Gauge".
-            
+
             ## Sampling Method
             ###### This is the technique the AI uses to create your image. Each option is a different approach, like choosing between pencils, markers, or paint. You don't need to worry too much about this; the default setting is usually the best choice for most users.
 
@@ -526,15 +528,16 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme_5') as dalle:
             ### Remember, these settings are all about giving you control over the image generation process. Feel free to experiment and see what each one does. And if you're ever in doubt, the default settings are a great place to start. Happy creating!
             """
             )
+
     # Row containing the 'Run' button to trigger the image generation
     with gr.Row():
         text_button = gr.Button("Run", variant='primary', elem_id="gen-button")
     # Row for displaying the generated image output
     with gr.Row():
         image_output = gr.Image(type="pil", label="Image Output", elem_id="gallery")
-        
+
     # Set up button click event to call the query function
-    text_button.click(query, inputs=[text_prompt, model, negative_prompt, steps, cfg, method, seed, strength, width, height], outputs=image_output)
+    text_button.click(query, inputs=[text_prompt, model, custom_lora, negative_prompt, steps, cfg, method, seed, strength, width, height], outputs=image_output)
 
 print("Launching Gradio interface...")  # Debug log
 # Launch the Gradio interface without showing the API or sharing externally