Adjust the layout

app.py

@@ -112,12 +112,12 @@ 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, batch_size, sr_option]}
+    d = {"data":[raw_text, int(batch_size), sr_option]}
     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]
     images = []
-    for i in range(batch_size):
+    for i in range(int(batch_size)):
         # print(content[i])
         images.append(base2picture(content[i]))        
 
@@ -302,8 +302,8 @@ if __name__ == "__main__":
                     #             rounded=(True, True, True, True),
                     #         )
 
-                    text.submit(request_images, inputs=[text, class_draw, style_draw, int(sample_size), sr_option], outputs=gallery)
-                    btn.click(request_images, inputs=[text, class_draw, style_draw, int(sample_size), sr_option], outputs=gallery)
+                    text.submit(request_images, inputs=[text, class_draw, style_draw, sample_size, sr_option], 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])