Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
lixiang46
commited on
Commit
•
f8801dc
1
Parent(s):
1279496
add log
Browse files
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=
|
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 |
|