hiding masked_img

app.py

@@ -411,6 +411,10 @@ background: linear-gradient(135deg, #CCFED9, #95BAFF);
 #huggingface-space-header {
     display: none !important;
 }
+#masked-img {
+    background: #553566 !important;
+    color: white !important;
+    }
 #generate_btn {
     background: #553566 !important;
     color: white !important;
@@ -493,9 +497,9 @@ with image_blocks as demo:
                 with gr.Row():
                     prompt = gr.Textbox(placeholder="Description of garment ex) Short Sleeve Round Neck T-shirts", show_label=False, elem_id="prompt")
 
-        # with gr.Column():
-        #     # image_out = gr.Image(label="Output", elem_id="output-img", height=400)
-        #     masked_img = gr.Image(label="Masked image output", elem_id="masked-img",show_share_button=False)
+        with gr.Column():
+            # image_out = gr.Image(label="Output", elem_id="output-img", height=400)
+            masked_img = gr.Image(label="Masked image output", elem_id="masked-img",show_share_button=False)
        
         with gr.Column():
             # image_out = gr.Image(label="Output", elem_id="output-img", height=400)
@@ -505,7 +509,7 @@ with image_blocks as demo:
 
 
     with gr.Column():
-        try_button = gr.Button(value="Try-on")
+        try_button = gr.Button(value="Try-on",  elem_id="generate_btn")
         with gr.Accordion(label="Advanced Settings", open=False):
             with gr.Row():
                 denoise_steps = gr.Number(label="Denoising Steps", minimum=20, maximum=40, value=30, step=1)
@@ -513,7 +517,7 @@ with image_blocks as demo:
 
 
 
-    try_button.click(fn=start_tryon, inputs=[imgs, garm_img, prompt, is_checked,is_checked_crop, denoise_steps, seed], outputs=[image_out,_], api_name='tryon')
+    try_button.click(fn=start_tryon, inputs=[imgs, garm_img, prompt, is_checked,is_checked_crop, denoise_steps, seed], outputs=[image_out,masked-img], api_name='tryon')
 
 image_blocks.launch(auth=(HF_TEST_USER_NAME, HF_PASS), show_error=True)
 # image_blocks.launch(show_error=True)