Spaces:
Sleeping
Sleeping
lixiang46
commited on
Commit
•
2398aef
1
Parent(s):
aae94cd
check empty image
Browse files
app.py
CHANGED
@@ -10,6 +10,8 @@ import json
|
|
10 |
|
11 |
|
12 |
def start_tryon(person_img, garment_img, seed, randomize_seed):
|
|
|
|
|
13 |
if randomize_seed:
|
14 |
seed = random.randint(0, MAX_SEED)
|
15 |
encoded_person_img = cv2.imencode('.jpg', cv2.cvtColor(person_img, cv2.COLOR_RGB2BGR))[1].tobytes()
|
@@ -146,7 +148,6 @@ with gr.Blocks(css=css) as Tryon:
|
|
146 |
label=None
|
147 |
)
|
148 |
|
149 |
-
image1.clear()
|
150 |
|
151 |
image1.change(
|
152 |
fn = change_imgs,
|
|
|
10 |
|
11 |
|
12 |
def start_tryon(person_img, garment_img, seed, randomize_seed):
|
13 |
+
if person_img == None or garment_img == None:
|
14 |
+
return None, None, None, "Empty image"
|
15 |
if randomize_seed:
|
16 |
seed = random.randint(0, MAX_SEED)
|
17 |
encoded_person_img = cv2.imencode('.jpg', cv2.cvtColor(person_img, cv2.COLOR_RGB2BGR))[1].tobytes()
|
|
|
148 |
label=None
|
149 |
)
|
150 |
|
|
|
151 |
|
152 |
image1.change(
|
153 |
fn = change_imgs,
|