fixed sr bug

app.py

@@ -125,7 +125,7 @@ def request_images(raw_text, class_draw, style_draw, batch_size, sr_option, w, h
 
     return images
 
-def sr_request_images(img_str, idx):
+def sr_request_images(img_str, idx, w, h, seed):
     idx_map = {
             "图片1(img1)":0,
             "图片2(img2)":1,
@@ -136,7 +136,7 @@ def sr_request_images(img_str, idx):
     # image_data = re.sub('^data:image/.+;base64,', '', img_str[idx])
     image_data = img_str[idx]
     # print(image_data)
-    d = {"data":[image_data, 0, False, float(512), float(512), float(-1)]} # batch_size设置为0，sr_opt设置为False, 新加了3个参数w,h,seed,随便放就行，进去不会走这个路
+    d = {"data":[image_data, 0, False, w, h, seed]} # batch_size设置为0，sr_opt设置为False, 新加了3个参数w,h,seed,随便放就行，进去不会走这个路
     url = "http://flagart.baai.ac.cn/api/general/"
     r = requests.post(url, json=d, headers={"Content-Type": "application/json", "Accept": "*/*", "Accept-Encoding": "gzip, deflate, br", "Connection": "keep-alive"})
     result_text = r.text
@@ -315,7 +315,7 @@ if __name__ == "__main__":
                     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, w, h, seed], outputs=gallery)
                     # to do here
-                    output_txt2img_sr_btn.click(sr_request_images, inputs=[gallery, img_choices], outputs=[sr_gallery])
+                    output_txt2img_sr_btn.click(sr_request_images, inputs=[gallery, img_choices, w, h, seed], outputs=[sr_gallery])
 
                     sample_size.change(
                         fn=uifn.change_img_choices,