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 |
|