Update app.py

app.py

@@ -13,7 +13,7 @@ headers = {"Authorization": f"Bearer {API_TOKEN}"}
 
 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", "PixArt XL 2.0", "Disney Cartoon", "CleanLinearMix", "Waifu 1.4"]
 
-def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, seed=None, sampler="DPM++ 2M Karras"):
+def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=None):
     if prompt == None:
         return None
 
@@ -95,12 +95,14 @@ with gr.Blocks(css=css) as dalle:
 
     with gr.Tab("Расширенные настройки"):
         negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="Чего не должно быть на изображении", 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")
+        steps = gr.Slider(label="Sampling steps", value=30, minimum=1, maximum=100, step=1)
+        cfg = gr.Slider(label="CFG Scale", value=7, minimum=1, maximum=20, step=1)
 
     with gr.Row():
         text_button = gr.Button("Генерация", variant='primary', elem_id="gen-button")
     with gr.Row():
         image_output = gr.Image(type="pil", label="Изображение", elem_id="gallery")
         
-    text_button.click(query, inputs=[text_prompt, model, negative_prompt], outputs=image_output)
+    text_button.click(query, inputs=[text_prompt, model, negative_prompt, steps, cfg], outputs=image_output)
 
 dalle.launch(show_api=False)