add width,height,seed

app.py

@@ -98,7 +98,7 @@ def filter_content(raw_style: str):
     else :
         return raw_style[:i]
 
-def request_images(raw_text, class_draw, style_draw, batch_size, sr_option):
+def request_images(raw_text, class_draw, style_draw, batch_size, sr_option, w, h, seed):
     if filter_content(class_draw) != "国画":
         if filter_content(class_draw) != "通用":
             raw_text = raw_text + f"，{filter_content(class_draw)}"
@@ -114,7 +114,7 @@ def request_images(raw_text, class_draw, style_draw, batch_size, sr_option):
             raw_text = raw_text + "，国画"
         url = "http://flagart.baai.ac.cn/api/guohua/"
 
-    d = {"data":[raw_text, int(batch_size), sr_option]}
+    d = {"data":[raw_text, int(batch_size), sr_option, w, h, seed]}
     r = requests.post(url, json=d, headers={"Content-Type": "application/json", "Accept": "*/*", "Accept-Encoding": "gzip, deflate, br", "Connection": "keep-alive"})
     result_text = r.text
     content = json.loads(result_text)["data"][0]
@@ -270,6 +270,10 @@ if __name__ == "__main__":
                             #                         interactive=True,
                             #                         )
                             sample_size = gr.Radio(choices=["1","2","3","4"], value="1", show_label=True, label='生成数量(number)')
+                            seed = gr.Number(-1, label='seed', interactive=True)
+                        with gr.Row().style(mobile_collapse=False, equal_height=True):
+                            w = gr.Slider(512,1024,value=512, step=64, label="width")
+                            h = gr.Slider(512,1024,value=512, step=64, label="height")
                         with gr.Row(visible=False).style(mobile_collapse=False, equal_height=True):
                             sr_option = gr.Checkbox(value=False, label="是否使用超分(Whether to use super-resolution)")
 
@@ -308,7 +312,7 @@ if __name__ == "__main__":
                     #             rounded=(True, True, True, True),
                     #         )
 
-                    text.submit(request_images, inputs=[text, class_draw, style_draw, sample_size, sr_option], outputs=gallery)
+                    text.submit(request_images, inputs=[text, class_draw, style_draw, sample_size, sr_option, w, h, seed], outputs=gallery)
                     btn.click(request_images, inputs=[text, class_draw, style_draw, sample_size, sr_option], outputs=gallery)
                     # to do here
                     output_txt2img_sr_btn.click(sr_request_images, inputs=[gallery, img_choices], outputs=[sr_gallery])