SaMeT1693 commited on
Commit
e5ed969
·
verified ·
1 Parent(s): a7eb80d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -9
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 = "https://" + os.environ['tryon_url'] + "Submit"
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 = "https://" + os.environ['tryon_url'] + "Query?taskId=" + uuid
58
- response = requests.get(url, headers=headers, timeout=50)
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("Bağlanma hatası Too many users, please try again later ")
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 = "https://" + os.environ['tryon_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
  """)