Update app.py

app.py

@@ -1,4 +1,3 @@
-import grequests
 import os
 import gradio as gr
 import numpy as np
@@ -6,31 +5,32 @@ import random
 import torch
 import subprocess
 import time
+import requests
 import json
+import threading
+
 import base64
 from io import BytesIO
 from PIL import Image
 
 
-url = "http://54.91.63.201:80"
+url_SPR = "http://34.229.166.42:80"
+url_ICX = "http://54.221.56.4:80"
 
 print('=='*20)
 print(os.system("hostname -i"))
 
-def url_requests(req_list,sizeImg):
-    img_list = []
-    res_list = grequests.map(req_list)
-    for resp in res_list:
-        img_str = json.loads(resp.text)["img_str"]
-        print("location: ", json.loads(resp.text)["ip"])
-
-        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((sizeImg))
-        img_list.append(_img)
+def url_requests(url,sizeImg):
+    resp = requests.post(url, data=json.dumps(data))
+    img_str = json.loads(resp.text)["img_str"]
+
+    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((sizeImg))
+    return _img
+        
         
-    return img_list
 
 
 def img2img_generate(source_img, prompt, steps=25, strength=0.25, seed=42, guidance_scale=15):
@@ -46,16 +46,17 @@ def img2img_generate(source_img, prompt, steps=25, strength=0.25, seed=42, guida
     img_b64 = base64.b64encode(buffered.getvalue())
 
     data = {"source_img": img_b64.decode(), "prompt": prompt, "steps": steps,
-            "guidance_scale": guidance_scale, "seed": seed, "strength": strength}
+            "guidance_scale": guidance_scale, "seed": seed, "strength": strength,
+            "token": os.environ["access_token"]}
 
     start_time = time.time()
     req_list = [ 
         grequests.post(url, data=json.dumps(data)),
         grequests.post(url, data=json.dumps(data)),
     ]
-    img_list = url_requests(req_list,source_img.size)
+    img = url_requests(url,source_img.size)
     
-    return img_list
+    return img
 
 
 def txt2img_generate(prompt, steps=25, seed=42, guidance_scale=7.5):
@@ -124,6 +125,6 @@ with gr.Blocks(css=css) as demo:
                 result_image_4 = gr.Image(label="Result02", elem_id="img_1")
                 
     txt2img_button.click(fn=txt2img_generate, inputs=[prompt, inference_steps, seed, guidance_scale], outputs=[result_image_1, result_image_2], queue=False)
-    img2img_button.click(fn=img2img_generate, inputs=[source_img, prompt_2, inference_steps_2, strength, seed_2, guidance_scale_2], outputs=[result_image_3, result_image_4], queue=False)
-   
+    img2img_button.click(fn=img2img_generate, inputs=[url_SPR, source_img, prompt_2, inference_steps_2, strength, seed_2, guidance_scale_2], outputs=result_image_3, queue=False)
+    img2img_button.click(fn=img2img_generate, inputs=[url_CLX, source_img, prompt_2, inference_steps_2, strength, seed_2, guidance_scale_2], outputs=result_image_4, queue=False)
 demo.queue(default_enabled=False).launch(debug=True)