Update app.py

app.py

@@ -33,7 +33,7 @@ mmodels = {
 }
 timeout = 250
 
-def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, gpt=False):
+def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, gpt=False, height=512, width=512):
     if prompt == "" or prompt == None:
         return None
 
@@ -169,6 +169,9 @@ with gr.Blocks(css=css) as dalle:
             with gr.Tab("Расширенные настройки"):
                 with gr.Row():
                     negative_prompt = gr.Textbox(label="Исключения", 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")
+                with gr.Row():
+                    width = gr.Slider(label="Ширина", value=512, minimum=96, maximum=2000, step=16)
+                    height = gr.Slider(label="Высота", value=512, minimum=96, maximum=2000, step=16)
                 with gr.Row():
                     steps = gr.Slider(label="Количество шагов обработки", value=25, minimum=1, maximum=70, step=1)
                 with gr.Row():
@@ -195,6 +198,7 @@ with gr.Blocks(css=css) as dalle:
             with gr.Row():
                 image_output = gr.Image(type="pil", label="Изображение", elem_id="gallery", show_share_button=False)
         
-    text_button.click(query, inputs=[text_prompt, model, negative_prompt, steps, cfg, method, seed, gpt], outputs=image_output, concurrency_limit=24)
+    text_button.click(query, inputs=[text_prompt, model, negative_prompt, steps, cfg, method, seed, gpt, height, width], outputs=image_output, concurrency_limit=250)
+    text_prompt.submit(query, inputs=[text_prompt, model, negative_prompt, steps, cfg, method, seed, gpt, height, width], outputs=image_output, concurrency_limit=250)
 
 dalle.launch(show_api=False, share=False)