lixiang46 commited on
Commit
f8801dc
1 Parent(s): 1279496
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -7,9 +7,11 @@ import random
7
  import base64
8
  import requests
9
  import json
 
10
 
11
 
12
  def start_tryon(person_img, garment_img, seed, randomize_seed):
 
13
  if person_img is None or garment_img is None:
14
  return None, None, "Empty image"
15
  if randomize_seed:
@@ -32,7 +34,7 @@ def start_tryon(person_img, garment_img, seed, randomize_seed):
32
  }
33
 
34
  print("post")
35
- response = requests.post(url, headers=headers, data=json.dumps(data), timeout=60)
36
  print("response code", response.status_code)
37
  result_img = None
38
  if response.status_code == 200:
@@ -49,6 +51,8 @@ def start_tryon(person_img, garment_img, seed, randomize_seed):
49
  else:
50
  print(response.text)
51
  info = "URL error, pleace contact the admin"
 
 
52
 
53
  return result_img, seed, info
54
 
 
7
  import base64
8
  import requests
9
  import json
10
+ import time
11
 
12
 
13
  def start_tryon(person_img, garment_img, seed, randomize_seed):
14
+ start_time = time.time()
15
  if person_img is None or garment_img is None:
16
  return None, None, "Empty image"
17
  if randomize_seed:
 
34
  }
35
 
36
  print("post")
37
+ response = requests.post(url, headers=headers, data=json.dumps(data), timeout=50)
38
  print("response code", response.status_code)
39
  result_img = None
40
  if response.status_code == 200:
 
51
  else:
52
  print(response.text)
53
  info = "URL error, pleace contact the admin"
54
+ end_time = time.time()
55
+ print(f"time used: {end_time-start_time}")
56
 
57
  return result_img, seed, info
58