Update app.py

app.py

@@ -20,7 +20,7 @@ url_ICX = "http://54.221.56.4:80"
 print('=='*20)
 print(os.system("hostname -i"))
 
-def url_requests(url,sizeImg):
+def url_requests(url,data,sizeImg):
     resp = requests.post(url, data=json.dumps(data))
     img_str = json.loads(resp.text)["img_str"]
 
@@ -50,11 +50,8 @@ def img2img_generate(source_img, prompt, steps=25, strength=0.25, seed=42, guida
             "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 = url_requests(url,source_img.size)
+    
+    img = url_requests(url, data, source_img.size)
     
     return img
 
@@ -119,10 +116,13 @@ with gr.Blocks(css=css) as demo:
                 guidance_scale_2 = gr.inputs.Slider(1.0, 20.0, label='Guidance Scale - how much the prompt will influence the results', default=7.5, step=0.1)
                 strength = gr.inputs.Slider(0.0, 1.0, label='Strength - adding more noise to it the larger the strength', default=0.25, step=0.01)
                 img2img_button = gr.Button("Generate Image")
+                url_SPR = gr.Textbox(label='url_SPR', value="http://34.229.166.42:80", visible=False)
+                url_CLX = gr.Textbox(label='url_CLX', value="http://54.221.56.4:80", visible=False)
 
+                
             with gr.Column():
-                result_image_3 = gr.Image(label="Result01", elem_id="img_1")
-                result_image_4 = gr.Image(label="Result02", elem_id="img_1")
+                result_image_3 = gr.Image(label="Result01", elem_id="img_3")
+                result_image_4 = gr.Image(label="Result02", elem_id="img_4")
                 
     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=[url_SPR, source_img, prompt_2, inference_steps_2, strength, seed_2, guidance_scale_2], outputs=result_image_3, queue=False)