istudio commited on
Commit
95dcc3f
·
verified ·
1 Parent(s): d201505

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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['assets_'] + "Submit"
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['assets_'] + "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,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['assets_']
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
- "clothImage": encoded_garment_img,
116
- "humanImage": encoded_person_img,
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