Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ def tryon(person_img, garment_img, seed, randomize_seed):
|
|
21 |
encoded_garment_img = cv2.imencode('.jpg', cv2.cvtColor(garment_img, cv2.COLOR_RGB2BGR))[1].tobytes()
|
22 |
encoded_garment_img = base64.b64encode(encoded_garment_img).decode('utf-8')
|
23 |
|
24 |
-
url = "
|
25 |
token = os.environ['token']
|
26 |
cookie = os.environ['Cookie']
|
27 |
referer = os.environ['referer']
|
@@ -54,8 +54,8 @@ def tryon(person_img, garment_img, seed, randomize_seed):
|
|
54 |
err_log = ""
|
55 |
for i in range(Max_Retry):
|
56 |
try:
|
57 |
-
url = "
|
58 |
-
response = requests.get(url, headers=headers, timeout=
|
59 |
# print("get response code", response.status_code)
|
60 |
if response.status_code == 200:
|
61 |
result = response.json()['result']
|
@@ -88,11 +88,9 @@ def tryon(person_img, garment_img, seed, randomize_seed):
|
|
88 |
if info == "":
|
89 |
err_log = f"No image after {Max_Retry} retries"
|
90 |
info = "Too many users, please try again later"
|
91 |
-
|
92 |
if info != "Success":
|
93 |
print(f"Error Log: {err_log}")
|
94 |
-
gr.Warning("
|
95 |
-
|
96 |
|
97 |
return result_img, seed, info
|
98 |
|
@@ -107,7 +105,7 @@ def start_tryon(person_img, garment_img, seed, randomize_seed):
|
|
107 |
encoded_garment_img = cv2.imencode('.jpg', cv2.cvtColor(garment_img, cv2.COLOR_RGB2BGR))[1].tobytes()
|
108 |
encoded_garment_img = base64.b64encode(encoded_garment_img).decode('utf-8')
|
109 |
|
110 |
-
url = "
|
111 |
token = os.environ['token']
|
112 |
cookie = os.environ['Cookie']
|
113 |
referer = os.environ['referer']
|
@@ -260,9 +258,7 @@ with gr.Blocks(css=css) as Tryon:
|
|
260 |
<div> </div>
|
261 |
<br>
|
262 |
<div>
|
263 |
-
|
264 |
Virtual try-on examples in pairs of person and garment images
|
265 |
-
|
266 |
</div>
|
267 |
</div>
|
268 |
""")
|
|
|
21 |
encoded_garment_img = cv2.imencode('.jpg', cv2.cvtColor(garment_img, cv2.COLOR_RGB2BGR))[1].tobytes()
|
22 |
encoded_garment_img = base64.b64encode(encoded_garment_img).decode('utf-8')
|
23 |
|
24 |
+
url = "http://" + os.environ['tryon_url'] + "Submit"
|
25 |
token = os.environ['token']
|
26 |
cookie = os.environ['Cookie']
|
27 |
referer = os.environ['referer']
|
|
|
54 |
err_log = ""
|
55 |
for i in range(Max_Retry):
|
56 |
try:
|
57 |
+
url = "http://" + os.environ['tryon_url'] + "Query?taskId=" + uuid
|
58 |
+
response = requests.get(url, headers=headers, timeout=20)
|
59 |
# print("get response code", response.status_code)
|
60 |
if response.status_code == 200:
|
61 |
result = response.json()['result']
|
|
|
88 |
if info == "":
|
89 |
err_log = f"No image after {Max_Retry} retries"
|
90 |
info = "Too many users, please try again later"
|
|
|
91 |
if info != "Success":
|
92 |
print(f"Error Log: {err_log}")
|
93 |
+
gr.Warning("Too many users, please try again later")
|
|
|
94 |
|
95 |
return result_img, seed, info
|
96 |
|
|
|
105 |
encoded_garment_img = cv2.imencode('.jpg', cv2.cvtColor(garment_img, cv2.COLOR_RGB2BGR))[1].tobytes()
|
106 |
encoded_garment_img = base64.b64encode(encoded_garment_img).decode('utf-8')
|
107 |
|
108 |
+
url = "http://" + os.environ['tryon_url']
|
109 |
token = os.environ['token']
|
110 |
cookie = os.environ['Cookie']
|
111 |
referer = os.environ['referer']
|
|
|
258 |
<div> </div>
|
259 |
<br>
|
260 |
<div>
|
|
|
261 |
Virtual try-on examples in pairs of person and garment images
|
|
|
262 |
</div>
|
263 |
</div>
|
264 |
""")
|