Update app.py

app.py

@@ -118,6 +118,6 @@ with gr.Blocks(css=css) as dalle:
     with gr.Row():
         image_output = gr.Image(type="pil", label="Изображение", elem_id="gallery")
         
-    text_button.click(query, inputs=[text_prompt, model, negative_prompt, steps, cfg, method, seed], outputs=image_output)
+    text_button.click(query, inputs=[text_prompt, model, negative_prompt, steps, cfg, method, seed, custom], outputs=image_output)
 
 dalle.launch(show_api=False, share=False)