Update run/gradio_ootd.py

run/gradio_ootd.py

@@ -191,13 +191,13 @@ with block:
         with gr.Column():
             vton_img_dc = gr.Image(label="Model", sources='upload', type="filepath", height=384, value=model_dc)
             example = gr.Examples(
-                label="",
+                label="Select for Upper/Lower Body",
                 inputs=vton_img_dc,
                 examples_per_page=7,
                 examples=[
-                    # os.path.join(example_path, 'model/model_8.png')
+                    os.path.join(example_path, 'model/model_8.png'),
                 #     os.path.join(example_path, 'model/049447_0.jpg'),
-                #     os.path.join(example_path, 'model/049713_0.jpg'),
+                    os.path.join(example_path, 'model/049713_0.jpg')
                 #     os.path.join(example_path, 'model/051482_0.jpg'),
                 #     os.path.join(example_path, 'model/051918_0.jpg'),
                 #     os.path.join(example_path, 'model/051962_0.jpg'),
@@ -205,14 +205,14 @@ with block:
                 ]
                 )
             example = gr.Examples(
-                label="",
+                label="Select for Full Body Dress",
                 inputs=vton_img_dc,
                 examples_per_page=7,
                 examples=[
-                    # os.path.join(example_path, 'model/model_9.png')
+                    os.path.join(example_path, 'model/model_9.png'),
                 #     os.path.join(example_path, 'model/052767_0.jpg'),
                 #     os.path.join(example_path, 'model/052472_0.jpg'),
-                #     os.path.join(example_path, 'model/053514_0.jpg'),
+                    os.path.join(example_path, 'model/053514_0.jpg')
                 #     os.path.join(example_path, 'model/053228_0.jpg'),
                 #     os.path.join(example_path, 'model/052964_0.jpg'),
                 #     os.path.join(example_path, 'model/053700_0.jpg'),
@@ -226,6 +226,18 @@ with block:
                 inputs=garm_img_dc,
                 examples_per_page=7,
                 examples=[
+                    os.path.join(example_path,'garment/01260_00.jpg '),
+                    os.path.join(example_path,'garment/01430_00.jpg '),
+                    os.path.join(example_path,'garment/02783_00.jpg '),
+                    os.path.join(example_path,'garment/03751_00.jpg '),
+                    os.path.join(example_path,'garment/06429_00.jpg '),
+                    os.path.join(example_path,'garment/06802_00.jpg '),
+                    os.path.join(example_path,'garment/07429_00.jpg '),
+                    os.path.join(example_path,'garment/08348_00.jpg '),
+                    os.path.join(example_path,'garment/09933_00.jpg '),
+                    os.path.join(example_path,'garment/11028_00.jpg '),
+                    os.path.join(example_path,'garment/11351_00.jpg '),
+                    os.path.join(example_path,'garment/11791_00.jpg '),
                     os.path.join(example_path, 'garment/048554_1.jpg'),
                     os.path.join(example_path, 'garment/049920_1.jpg'),
                     os.path.join(example_path, 'garment/049965_1.jpg'),
@@ -264,13 +276,13 @@ with block:
             result_gallery_dc = gr.Gallery(label='Output', 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)
-        # 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)
+        # 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)
+        # # 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)
         
-    ips_dc = [vton_img_dc, garm_img_dc, category_dc, n_samples_dc, n_steps_dc, image_scale_dc, seed_dc]
+    ips_dc = [vton_img_dc, garm_img_dc, category_dc, 1, 20, 2.0, -1]
     run_button_dc.click(fn=process_dc, inputs=ips_dc, outputs=[result_gallery_dc])