Upload gradio_ootd.py

run/gradio_ootd.py

@@ -121,14 +121,14 @@ def process_dc(vton_img, garm_img, category, n_samples, n_steps, image_scale, se
 block = gr.Blocks().queue()
 with block:
     with gr.Row():
-        gr.Markdown("# OOTDiffusion Demo")
+        gr.Markdown("# 一键换装功能")
     with gr.Row():
-        gr.Markdown("## Half-body")
+        gr.Markdown("## 半身")
     with gr.Row():
-        gr.Markdown("***Support upper-body garments***")
+        gr.Markdown("***支持上衣服装***")
     with gr.Row():
         with gr.Column():
-            vton_img = gr.Image(label="Model", sources='upload', type="filepath", height=384, value=model_hd)
+            vton_img = gr.Image(label="模特", sources='upload', type="filepath", height=384, value=model_hd)
             example = gr.Examples(
                 inputs=vton_img,
                 examples_per_page=14,
@@ -149,7 +149,7 @@ with block:
                     os.path.join(example_path, 'model/01861_00.jpg'),
                 ])
         with gr.Column():
-            garm_img = gr.Image(label="Garment", sources='upload', type="filepath", height=384, value=garment_hd)
+            garm_img = gr.Image(label="服装", sources='upload', type="filepath", height=384, value=garment_hd)
             example = gr.Examples(
                 inputs=garm_img,
                 examples_per_page=14,
@@ -170,28 +170,28 @@ with block:
                     os.path.join(example_path, 'garment/04825_00.jpg'),
                 ])
         with gr.Column():
-            result_gallery = gr.Gallery(label='Output', show_label=False, elem_id="gallery", preview=True, scale=1)   
+            result_gallery = gr.Gallery(label='输出', show_label=False, elem_id="gallery", preview=True, scale=1)   
     with gr.Column():
-        run_button = gr.Button(value="Run")
-        n_samples = gr.Slider(label="Images", minimum=1, maximum=4, value=1, step=1)
-        n_steps = gr.Slider(label="Steps", minimum=20, maximum=40, value=20, step=1)
+        run_button = gr.Button(value="运行")
+        n_samples = gr.Slider(label="出图数", minimum=1, maximum=4, value=1, step=1)
+        n_steps = gr.Slider(label="步数", minimum=20, maximum=40, value=20, step=1)
         # scale = gr.Slider(label="Scale", minimum=1.0, maximum=12.0, value=5.0, step=0.1)
-        image_scale = gr.Slider(label="Guidance scale", minimum=1.0, maximum=5.0, value=2.0, step=0.1)
-        seed = gr.Slider(label="Seed", minimum=-1, maximum=2147483647, step=1, value=-1)
+        image_scale = gr.Slider(label="引导相关性", minimum=1.0, maximum=5.0, value=2.0, step=0.1)
+        seed = gr.Slider(label="随机种子", minimum=-1, maximum=2147483647, step=1, value=-1)
         
     ips = [vton_img, garm_img, n_samples, n_steps, image_scale, seed]
     run_button.click(fn=process_hd, inputs=ips, outputs=[result_gallery])
 
 
     with gr.Row():
-        gr.Markdown("## Full-body")
+        gr.Markdown("## 全身")
     with gr.Row():
-        gr.Markdown("***Support upper-body/lower-body/dresses; garment category must be paired!!!***")
+        gr.Markdown("***服装类别必须匹配!!!***")
     with gr.Row():
         with gr.Column():
-            vton_img_dc = gr.Image(label="Model", sources='upload', type="filepath", height=384, value=model_dc)
+            vton_img_dc = gr.Image(label="模特", sources='upload', type="filepath", height=384, value=model_dc)
             example = gr.Examples(
-                label="Examples (upper-body/lower-body)",
+                label="示例 (上半身/下半身)",
                 inputs=vton_img_dc,
                 examples_per_page=7,
                 examples=[
@@ -204,7 +204,7 @@ with block:
                     os.path.join(example_path, 'model/049205_0.jpg'),
                 ])
             example = gr.Examples(
-                label="Examples (dress)",
+                label="示例 (连衣裙)",
                 inputs=vton_img_dc,
                 examples_per_page=7,
                 examples=[
@@ -217,10 +217,10 @@ with block:
                     os.path.join(example_path, 'model/053700_0.jpg'),
                 ])
         with gr.Column():
-            garm_img_dc = gr.Image(label="Garment", sources='upload', type="filepath", height=384, value=garment_dc)
-            category_dc = gr.Dropdown(label="Garment category (important option!!!)", choices=["Upper-body", "Lower-body", "Dress"], value="Upper-body")
+            garm_img_dc = gr.Image(label="服装", sources='upload', type="filepath", height=384, value=garment_dc)
+            category_dc = gr.Dropdown(label="服装类别", choices=["Upper-body", "Lower-body", "Dress"], value="Upper-body")
             example = gr.Examples(
-                label="Examples (upper-body)",
+                label="示例 (上半身)",
                 inputs=garm_img_dc,
                 examples_per_page=7,
                 examples=[
@@ -233,7 +233,7 @@ with block:
                     os.path.join(example_path, 'garment/050105_1.jpg'),
                 ])
             example = gr.Examples(
-                label="Examples (lower-body)",
+                label="示例 (下半身)",
                 inputs=garm_img_dc,
                 examples_per_page=7,
                 examples=[
@@ -246,7 +246,7 @@ with block:
                     os.path.join(example_path, 'garment/051412_1.jpg'),
                 ])
             example = gr.Examples(
-                label="Examples (dress)",
+                label="示例 (连衣裙)",
                 inputs=garm_img_dc,
                 examples_per_page=7,
                 examples=[
@@ -259,14 +259,14 @@ with block:
                     os.path.join(example_path, 'garment/052234_1.jpg'),
                 ])
         with gr.Column():
-            result_gallery_dc = gr.Gallery(label='Output', show_label=False, elem_id="gallery", preview=True, scale=1)   
+            result_gallery_dc = gr.Gallery(label='输出', show_label=False, elem_id="gallery", preview=True, scale=1)   
     with gr.Column():
-        run_button_dc = gr.Button(value="Run")
-        n_samples_dc = gr.Slider(label="Images", minimum=1, maximum=4, value=1, step=1)
-        n_steps_dc = gr.Slider(label="Steps", minimum=20, maximum=40, value=20, step=1)
+        run_button_dc = gr.Button(value="运行")
+        n_samples_dc = gr.Slider(label="出图数", minimum=1, maximum=4, value=1, step=1)
+        n_steps_dc = gr.Slider(label="步数", minimum=20, maximum=40, value=20, step=1)
         # scale_dc = gr.Slider(label="Scale", minimum=1.0, maximum=12.0, value=5.0, step=0.1)
-        image_scale_dc = gr.Slider(label="Guidance scale", minimum=1.0, maximum=5.0, value=2.0, step=0.1)
-        seed_dc = gr.Slider(label="Seed", minimum=-1, maximum=2147483647, step=1, value=-1)
+        image_scale_dc = gr.Slider(label="引导相关性", minimum=1.0, maximum=5.0, value=2.0, step=0.1)
+        seed_dc = gr.Slider(label="随机种子", minimum=-1, maximum=2147483647, step=1, value=-1)
         
     ips_dc = [vton_img_dc, garm_img_dc, category_dc, n_samples_dc, n_steps_dc, image_scale_dc, seed_dc]
     run_button_dc.click(fn=process_dc, inputs=ips_dc, outputs=[result_gallery_dc])