Update app.py

app.py

@@ -17,7 +17,7 @@ url = "http://54.91.63.201:80"
 print('=='*20)
 print(os.system("hostname -i"))
 
-def url_requests(req_list):
+def url_requests(req_list,sizeImg):
     img_list = []
     res_list = grequests.map(req_list)
     for resp in res_list:
@@ -27,7 +27,7 @@ def url_requests(req_list):
         img_byte = base64.b64decode(img_str)
         img_io = BytesIO(img_byte)  # convert image to file-like object
         img = Image.open(img_io)   # img is now PIL Image object
-        _img = img.resize((source_img.size))
+        _img = img.resize((sizeImg))
         img_list.append(_img)
         
     return img_list
@@ -53,7 +53,7 @@ def img2img_generate(source_img, prompt, steps=25, strength=0.75, seed=42, guida
         grequests.post(url, data=json.dumps(data)),
         grequests.post(url, data=json.dumps(data)),
     ]
-    img_list = url_requests(req_list)
+    img_list = url_requests(req_list,source_img.size)
     
     return img_list