Update app.py

app.py

@@ -193,35 +193,33 @@ def start_tryon(person_img, pose_img, mask_img, cloth_img, garment_des, denoise_
 
     return images
 
+
 # Gradio interface for the virtual try-on model
 image_blocks = gr.Blocks().queue()
 
 with image_blocks as demo:
-    gr.Markdown("## SmartLuga ")
+    gr.Markdown("## SmartLuga")
+    gr.Markdown("Virtual Try-on with your image and garment image. Check out the [source codes](https://github.com/yisol/IDM-VTON) and the [model](https://huggingface.co/yisol/IDM-VTON)")
+    
     with gr.Row():
         with gr.Column():
-            imgs = gr.ImageEditor(sources='upload', type="pil", label='Human Image', interactive=True)
-            with gr.Row():
-                is_checked_crop = gr.Checkbox(label="Use auto-crop & resizing", value=False)
+            person_img = gr.Image(label='Person Image', sources='upload', type="pil")
+            pose_img = gr.Image(label='Pose Image', sources='upload', type="pil")
+            mask_img = gr.Image(label='Mask Image', sources='upload', type="pil")
 
         with gr.Column():
-            garm_img = gr.Image(label="Garment", sources='upload', type="pil")
-            with gr.Row(elem_id="prompt-container"):
-                prompt = gr.Textbox(placeholder="Description of garment ex) Short Sleeve Round Neck T-shirts", show_label=False, elem_id="prompt")
+            cloth_img = gr.Image(label='Garment Image', sources='upload', type="pil")
+            garment_des = gr.Textbox(placeholder="Description of garment ex) Short Sleeve Round Neck T-shirts", label="Garment Description")
 
         with gr.Column():
-            masked_img = gr.Image(label="Masked image output", elem_id="masked-img", show_share_button=False)
+            denoise_steps = gr.Number(label="Denoising Steps", minimum=20, maximum=40, value=30, step=1)
+            seed = gr.Number(label="Seed", minimum=-1, maximum=2147483647, step=1, value=42)
 
         with gr.Column():
-            image_out = gr.Image(label="Output", elem_id="output-img", show_share_button=False)
-
-    with gr.Column():
-        try_button = gr.Button(value="Try-on")
-        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)
-                seed = gr.Number(label="Seed", minimum=-1, maximum=2147483647, step=1, value=42)
+            masked_img = gr.Image(label="Masked Image Output", elem_id="masked-img", show_share_button=False)
+            image_out = gr.Image(label="Output Image", elem_id="output-img", show_share_button=False)
 
-    try_button.click(fn=start_tryon, inputs=[imgs, garm_img, prompt, denoise_steps, seed], outputs=[image_out, masked_img], api_name='tryon')
+    try_button = gr.Button(value="Try-on")
+    try_button.click(fn=start_tryon, inputs=[person_img, pose_img, mask_img, cloth_img, garment_des, denoise_steps, seed], outputs=[image_out, masked_img], api_name='tryon')
 
 image_blocks.launch()