Update virtualtryon_tab.py

virtualtryon_tab.py

@@ -1,31 +1,31 @@
-from src.rep_api import virtual_try_on
-import gradio as gr 
-
-
-def create_virtualtryon_tab():
-    with gr.TabItem("Virtual Try-on",id="virtual_tryon"):
-        with gr.Row():
-            with gr.Column():
-                human_img =  gr.Image(label="Human Image")
-                crop = gr.Checkbox(label="Crop", value=False)
-                seed = gr.Number(label="Seed", value=42)
-                steps = gr.Number(label="Steps", value=30)
-            with gr.Column():
-                garm_img = gr.Image(label="Garment Image")
-                category = gr.Dropdown(["upper_body", "lower_body", "full_body"], label="Category", value="upper_body")
-                garment_des = gr.Textbox(label="Garment Description")
-        with gr.Row():
-            tryon_btn = gr.Button("Try-on")
-        with gr.Row():
-            tryon_out = gr.Image(label="Output", type="filepath")
-        with gr.Row():
-           examples = gr.Examples(
-                examples=[
-                    [ False, 42, 30, "upper_body", "Test_images/pink_jumper.png", "Test_images/person_1.jpg","pink jumper"],
-                    [ True, 100, 50, "upper_body", "Test_images/Suit_2.png", "Test_images/person_2.jpg","Suit"],
-                  #  [ False, 7, 20, "full_body", "path/to/garment_img3.jpg", "path/to/human_img3.jpg","Black Suit"]
-                ],
-                inputs=[ crop, seed, steps, category, garm_img,human_img, garment_des]
-            )
-
+from src.rep_api import virtual_try_on
+import gradio as gr 
+
+
+def create_virtualtryon_tab():
+    with gr.TabItem("Virtual Try-on",id="virtual_tryon"):
+        with gr.Row():
+            with gr.Column():
+                human_img =  gr.Image(label="Human Image")
+                crop = gr.Checkbox(label="Crop", value=False)
+                seed = gr.Number(label="Seed", value=42)
+                steps = gr.Number(label="Steps", value=30)
+            with gr.Column():
+                garm_img = gr.Image(label="Garment Image")
+                category = gr.Dropdown(["upper_body", "lower_body", "full_body"], label="Category", value="upper_body")
+                garment_des = gr.Textbox(label="Garment Description")
+        with gr.Row():
+            tryon_btn = gr.Button("Try-on")
+        with gr.Row():
+            tryon_out = gr.Image(label="Output", type="filepath")
+        with gr.Row():
+           examples = gr.Examples(
+                examples=[
+                    [ False, 42, 30, "upper_body", "Test_images/pink_jumper.png", "Test_images/person_1.jpg","pink jumper"],
+                    [ True, 100, 30, "upper_body", "Test_images/Suit_2.png", "Test_images/person_2.jpg","Suit"],
+                  #  [ False, 7, 20, "full_body", "path/to/garment_img3.jpg", "path/to/human_img3.jpg","Black Suit"]
+                ],
+                inputs=[ crop, seed, steps, category, garm_img,human_img, garment_des]
+            )
+
     tryon_btn.click(virtual_try_on, inputs=[crop, seed, steps, category,  garm_img, human_img,garment_des],outputs=tryon_out)