Update app.py

app.py

@@ -11,11 +11,11 @@ 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", "DALL-E 3 XL", "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"]
+models_list = ["AbsoluteReality 1.8.1", "DALL-E 3 XL", "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", "Custom"]
 
 # PLEASE ❤ like ❤ this space. Please like me. I am 12 years old, one of my projects is: https://ai-hub.rf.gd . I live in Russia, I don't know English very well. Therefore, I apologize that there is only Russian here, but I think it will not be difficult to translate all this. (For example, using gpt)
 
-def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1):
+def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, custom=None):
     if prompt == "" or prompt == None:
         return None
 
@@ -64,6 +64,8 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
         API_URL = "https://api-inference.huggingface.co/models/goofyai/disney_style_xl"
     if model == 'CleanLinearMix':
         API_URL = "https://api-inference.huggingface.co/models/digiplay/CleanLinearMix_nsfw"
+    if model == 'custom':
+        API_URL == f"https://api-inference.huggingface.co/models/{custom}"
     
     payload = {
         "inputs": prompt,
@@ -92,6 +94,9 @@ with gr.Blocks(css=css) as dalle:
                     text_prompt = gr.Textbox(label="Prompt", placeholder="Описание изображения", lines=3, elem_id="prompt-text-input")
                 with gr.Row():
                     model = gr.Radio(label="Модель", value="DALL-E 3 XL", choices=models_list)
+                    with gr.Accordion(label="Custom", open=False):
+                        custom = gr.Textbox(show_label=False, placeholder="autor/model — openskyml/dalle-3-xl")
+
                 
 
     with gr.Tab("Расширенные настройки"):