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 = "http://" + os.environ['
|
25 |
token = os.environ['token']
|
26 |
cookie = os.environ['Cookie']
|
27 |
referer = os.environ['referer']
|
@@ -54,7 +54,7 @@ def tryon(person_img, garment_img, seed, randomize_seed):
|
|
54 |
err_log = ""
|
55 |
for i in range(Max_Retry):
|
56 |
try:
|
57 |
-
url = "http://" + os.environ['
|
58 |
response = requests.get(url, headers=headers, timeout=20)
|
59 |
# print("get response code", response.status_code)
|
60 |
if response.status_code == 200:
|
@@ -105,15 +105,15 @@ def start_tryon(person_img, garment_img, seed, randomize_seed):
|
|
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['
|
109 |
token = os.environ['token']
|
110 |
cookie = os.environ['Cookie']
|
111 |
referer = os.environ['referer']
|
112 |
|
113 |
headers = {'Content-Type': 'application/json', 'token': token, 'Cookie': cookie, 'referer': referer}
|
114 |
data = {
|
115 |
-
"
|
116 |
-
"
|
117 |
"seed": seed
|
118 |
}
|
119 |
|
|
|
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:
|
|
|
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']
|
112 |
|
113 |
headers = {'Content-Type': 'application/json', 'token': token, 'Cookie': cookie, 'referer': referer}
|
114 |
data = {
|
115 |
+
"imagesImage": encoded_garment_img,
|
116 |
+
"interpolationImage": encoded_person_img,
|
117 |
"seed": seed
|
118 |
}
|
119 |
|