Update app.py

app.py

@@ -190,7 +190,7 @@ def start_tryon(dict,garm_img,garment_des,cloth_type,is_checked,is_checked_crop,
         human_img_orig.paste(out_img, (int(left), int(top)))    
         return human_img_orig, mask_gray
     else:
-        return images[0], mask_gray
+        return garm_img, images[0], mask_gray
     # return images[0], mask_gray
 
 
@@ -259,6 +259,7 @@ with image_blocks as demo:
             # image_out = gr.Image(label="Output", elem_id="output-img", height=400)
             image_out = gr.Image(label="Output", elem_id="output-img",show_share_button=False)
             with gr.Accordion("Debug Info", open=False):
+                image_in = gr.Image(label="Midway Image", elem_id="midway-img",show_share_button=False)
                 masked_img = gr.Image(label="Masked image output", elem_id="masked-img",show_share_button=False)
 
 
@@ -273,7 +274,7 @@ with image_blocks as demo:
 
 
     try_button.click(fn=main_, inputs=[imgs,topwear_image,topwear_desc,bottomwear_image,bottomwear_desc,dress_image,dress_desc,is_checked,is_checked_crop,denoise_steps,seed],
-                     outputs=[image_out, masked_img], api_name='tryon')
+                     outputs=[image_in, image_out, masked_img], api_name='tryon')