Spaces:

Kwai-Kolors
/

Kolors-Virtual-Try-On

Running on CPU Upgrade

lixiang46 commited on Aug 12

Commit

0f0fc4b

•

1 Parent(s): 2398aef

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ import json
 def start_tryon(person_img, garment_img, seed, randomize_seed):
-    if person_img == None or garment_img == None:
         return None, None, None, "Empty image"
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)

 def start_tryon(person_img, garment_img, seed, randomize_seed):
+    if person_img.all() == None or garment_img.all() == None:
         return None, None, None, "Empty image"
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)