Spaces:

Kwai-Kolors
/

Kolors-Virtual-Try-On

Running on CPU Upgrade

lixiang46 commited on Sep 3

Commit

22a00a3

•

1 Parent(s): 2231fef

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -82,7 +82,7 @@ def tryon(person_img, garment_img, seed, randomize_seed):
     get_end_time = time.time()
     print(f"get time used: {get_end_time-get_start_time}")
     print(f"all time used: {get_end_time-get_start_time+post_end_time-post_start_time}")
-    if result_img == None:
         err_log = f"No image after {Max_Retry} retries"
         info = "Too many users, please try again later"
     if info != "Success":

     get_end_time = time.time()
     print(f"get time used: {get_end_time-get_start_time}")
     print(f"all time used: {get_end_time-get_start_time+post_end_time-post_start_time}")
+    if info == "":
         err_log = f"No image after {Max_Retry} retries"
         info = "Too many users, please try again later"
     if info != "Success":