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, sampler="DPM++ 2M Karras", seed=None):
+def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, width, height):
     if prompt == "" or prompt == None:
         return None
 
@@ -72,7 +72,9 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
         "is_negative": is_negative,
         "steps": steps,
         "cfg_scale": cfg_scale,
-        "seed": seed if seed != -1 else random.randint(1, 1000000000)
+        "seed": seed if seed != -1 else random.randint(1, 1000000000),
+        "width": width,
+        "height": height
         }
 
     image_bytes = requests.post(API_URL, headers=headers, json=payload).content
@@ -107,12 +109,18 @@ with gr.Blocks(css=css) as dalle:
         with gr.Row():
             seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=1000000000, step=1)
 
+    with gr.Tab("Beta"):
+        with gr.Row():
+            width = gr.Slider(label="Ширина", value=512, minimum=15, maximum=1024, step=1)
+            height = gr.Slider(label="Длина", value=512, minimum=15, maximum=1024, 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, steps, cfg, method, seed], outputs=image_output)
+    text_button.click(query, inputs=[text_prompt, model, negative_prompt, steps, cfg, method, seed, width, height], outputs=image_output)
 
 dalle.launch(show_api=False)