Update app.py

app.py

@@ -32,7 +32,7 @@ mmodels = {
 timeout = 100
 # 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, strength=0.7, gpt=False):
+def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, gpt=False, width=1024, height=1024):
     if prompt == "" or prompt == None:
         return None
 
@@ -110,11 +110,11 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
         "cfg_scale": cfg_scale,
         "seed": seed if seed != -1 else random.randint(1, 1000000000),
         "strength": strength,
-        "width": 1024,
-        "height": 1024,
+        "width": width,
+        "height": height,
         "guidance_scale": cfg_scale,
         "num_inference_steps": steps,
-        "resolution": "512 x 1024",
+        "resolution": f"{width} x {height}",
         "negative_prompt": is_negative
         }
 
@@ -171,7 +171,8 @@ with gr.Blocks(css=css) as dalle:
 
     with gr.Tab("Информация"):
         with gr.Row():
-            gr.Markdown("Soon")
+            width = gr.Slider(label="Ширина", minimum=15, maximum=2000, value=1024, step=1)
+            height = gr.Slider(label="Высота", minimum=15, maximum=2000, value=1024, step=1)
 
     with gr.Tab("Информация"):
         with gr.Row():
@@ -187,6 +188,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, strength, gpt], outputs=image_output)
+    text_button.click(query, inputs=[text_prompt, model, negative_prompt, steps, cfg, method, seed, strength, gpt, width, height], outputs=image_output)
 
 dalle.launch(show_api=False, share=False)