Update run/gradio_ootd.py

run/gradio_ootd.py

@@ -119,15 +119,23 @@ def process_dc(vton_img, garm_img, category, n_samples, n_steps, image_scale, se
 
 block = gr.Blocks().queue()
 with block:
-vton_img = gr.Image(label="Model", sources='upload', type="filepath", height=384, value=model_hd)
-garm_img = gr.Image(label="Garment", sources='upload', type="filepath", height=384, value=garment_hd)
-result_gallery = gr.Gallery(label='Output', show_label=False, elem_id="gallery", preview=True, scale=1)   
-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)
-# 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)
+
+    with gr.Row(visible=False):
+        with gr.Column():
+            vton_img = gr.Image(label="Model", sources='upload', type="filepath", height=384, value=model_hd)
+           
+        with gr.Column(visible=False):
+            garm_img = gr.Image(label="Garment", sources='upload', type="filepath", height=384, value=garment_hd)
+           
+        with gr.Column(visible=False):
+            result_gallery = gr.Gallery(label='Output', show_label=False, elem_id="gallery", preview=True, scale=1)   
+    with gr.Column(visible=False):
+        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)
+        # 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)
         
     ips = [vton_img, garm_img, n_samples, n_steps, image_scale, seed]
     run_button.click(fn=process_hd, inputs=ips, outputs=[result_gallery])