remove mask

app.py

@@ -331,7 +331,7 @@ with image_blocks as demo:
         with gr.Column():
             garm_img = gr.Image(label="Garment", sources='upload', type="pil")
             try_button = gr.Button(value="Try-on")
-            masked_img = gr.Image(label="Masked image output", elem_id="masked-img",show_share_button=False)          
+            #masked_img = gr.Image(label="Masked image output", elem_id="masked-img",show_share_button=False, visi)          
             image_out = gr.Image(label="Output", elem_id="output-img",show_share_button=False)
             buy_link = gr.Textbox(label="URL of Selected Image", interactive=False, visible= False)
             buy_button = gr.Button(value="Like it? Click to buy")
@@ -339,8 +339,10 @@ with image_blocks as demo:
    
     fetch_button.click(fn=fetch_products, inputs=prompt, outputs=[image_gallery, url_display])
     image_gallery.select(select_image, [image_gallery, url_display], [garm_img, buy_link])
-    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')
+    #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')
     #buy_button.click(fn=None,inputs=buy_link,js=f"(buy_link) => {{ window.open(buy_link.substring(buy_link.indexOf('amazon.com')), '_blank');console.log(buy_link) }}")
+    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')
+
     buy_button.click(fn=None,inputs=buy_link, js=f'''(buy_link) => {{ 
         const clean_link = buy_link.includes('http') ? buy_link : 'https://' + buy_link;
         window.open(clean_link, '_blank');