Update app.py

app.py

@@ -46,7 +46,7 @@ def tryon(person_img, garment_img, seed, randomize_seed):
     post_end_time = time.time()
     print(f"post time used: {post_end_time-post_start_time}")
 
-    get_start_time =time.time()
+    get_start_time = time.time()
     time.sleep(9)
     Max_Retry = 10
     result_img = None
@@ -69,7 +69,7 @@ def tryon(person_img, garment_img, seed, randomize_seed):
                     raise gr.Error("Too many users, please try again later")
             else:
                 print(response.text)
-                info = "URL error, pleace contact the admin"
+                info = "URL error, please contact the admin"
         except requests.exceptions.ReadTimeout:
             print("timeout")
             info = "Too many users, please try again later"
@@ -117,17 +117,17 @@ def start_tryon(person_img, garment_img, seed, randomize_seed):
                 result_img = cv2.cvtColor(result_img, cv2.COLOR_RGB2BGR)
                 info = "Success"
             else:
-                info = "Try again latter"
+                info = "Try again later"
         else:
             print(response.text)
-            info = "URL error, pleace contact the admin"
+            info = "URL error, please contact the admin"
     except requests.exceptions.ReadTimeout:
         print("timeout")
         info = "Too many users, please try again later"
         raise gr.Error("Too many users, please try again later")
     except Exception as err:
-        print(f"其他错误: {err}")
-        info = "Error, pleace contact the admin"
+        print(f"Error: {err}")
+        info = "Error, please contact the admin"
     end_time = time.time()
     print(f"time used: {end_time-start_time}")
 
@@ -135,15 +135,7 @@ def start_tryon(person_img, garment_img, seed, randomize_seed):
 
 MAX_SEED = 999999
 
-example_path = os.path.join(os.path.dirname(__file__), 'assets')
-
-garm_list = os.listdir(os.path.join(example_path,"cloth"))
-garm_list_path = [os.path.join(example_path,"cloth",garm) for garm in garm_list]
-
-human_list = os.listdir(os.path.join(example_path,"human"))
-human_list_path = [os.path.join(example_path,"human",human) for human in human_list]
-
-css="""
+css = """
 #col-left {
     margin: 0 auto;
     max-width: 430px;
@@ -165,18 +157,12 @@ css="""
 }
 """
 
-def load_description(fp):
-    with open(fp, 'r', encoding='utf-8') as f:
-        content = f.read()
-    return content
-
 def change_imgs(image1, image2):
     return image1, image2
 
 with gr.Blocks(css=css) as Tryon:
-    gr.HTML(load_description("assets/title.md"))
     with gr.Row():
-        with gr.Column(elem_id = "col-left"):
+        with gr.Column(elem_id="col-left"):
             gr.HTML("""
             <div style="display: flex; justify-content: center; align-items: center; text-align: center; font-size: 20px;">
                 <div>
@@ -184,7 +170,7 @@ with gr.Blocks(css=css) as Tryon:
                 </div>
             </div>
             """)
-        with gr.Column(elem_id = "col-mid"):
+        with gr.Column(elem_id="col-mid"):
             gr.HTML("""
             <div style="display: flex; justify-content: center; align-items: center; text-align: center; font-size: 20px;">
                 <div>
@@ -192,7 +178,7 @@ with gr.Blocks(css=css) as Tryon:
                 </div>
             </div>
             """)
-        with gr.Column(elem_id = "col-right"):
+        with gr.Column(elem_id="col-right"):
             gr.HTML("""
             <div style="display: flex; justify-content: center; align-items: center; text-align: center; font-size: 20px;">
                 <div>
@@ -201,22 +187,11 @@ with gr.Blocks(css=css) as Tryon:
             </div>
             """)
     with gr.Row():
-        with gr.Column(elem_id = "col-left"):
+        with gr.Column(elem_id="col-left"):
             imgs = gr.Image(label="Person image", sources='upload', type="numpy")
-            # category = gr.Dropdown(label="Garment category", choices=['upper_body', 'lower_body', 'dresses'],  value="upper_body")
-            example = gr.Examples(
-                inputs=imgs,
-                examples_per_page=12,
-                examples=human_list_path
-            )
-        with gr.Column(elem_id = "col-mid"):
+        with gr.Column(elem_id="col-mid"):
             garm_img = gr.Image(label="Garment image", sources='upload', type="numpy")
-            example = gr.Examples(
-                inputs=garm_img,
-                examples_per_page=12,
-                examples=garm_list_path
-            )
-        with gr.Column(elem_id = "col-right"):
+        with gr.Column(elem_id="col-right"):
             image_out = gr.Image(label="Result", show_share_button=False)
             with gr.Row():
                 seed = gr.Slider(
@@ -230,14 +205,11 @@ with gr.Blocks(css=css) as Tryon:
             with gr.Row():
                 seed_used = gr.Number(label="Seed used")
                 result_info = gr.Text(label="Response")
-            # try_button = gr.Button(value="Run", elem_id="button")
             test_button = gr.Button(value="Run", elem_id="button")
 
+    test_button.click(fn=tryon, inputs=[imgs, garm_img, seed, randomize_seed], outputs=[image_out, seed_used, result_info], api_name='tryon', concurrency_limit=10)
 
-    # try_button.click(fn=start_tryon, inputs=[imgs, garm_img, seed, randomize_seed], outputs=[image_out, seed_used, result_info], api_name='tryon',concurrency_limit=10)
-    test_button.click(fn=tryon, inputs=[imgs, garm_img, seed, randomize_seed], outputs=[image_out, seed_used, result_info], api_name='tryon',concurrency_limit=10)
-
-    with gr.Column(elem_id = "col-showcase"):
+    with gr.Column(elem_id="col-showcase"):
         gr.HTML("""
         <div style="display: flex; justify-content: center; align-items: center; text-align: center; font-size: 20px;">
             <div> </div>
@@ -248,6 +220,4 @@ with gr.Blocks(css=css) as Tryon:
         </div>
         """)
 
-# ip = requests.get('http://ifconfig.me/ip', timeout=1).text.strip()
-# print("ip address", ip)
-Tryon.queue(max_size = 20).launch(max_threads = 5)
+Tryon.queue(max_size=20).launch(max_threads=5)