Upload app.py

app.py

@@ -19,7 +19,7 @@ client = Client(
 def virtual_tryon(
     person_path,
     garment_path,
-    vt_garment_type="upper_body",
+    garment_type,
 ):
     # 2) Wrap file inputs so Gradio client uploads them correctly
     person_file = handle_file(
@@ -27,6 +27,7 @@ def virtual_tryon(
     )  # handle_file uploads the image :contentReference[oaicite:6]{index=6}
     garment_file = handle_file(garment_path)
 
+    
     # 3) Build inputs in the exact order shown on the “Use via API” page :contentReference[oaicite:7]{index=7}
 
     # 4) Call the named endpoint with handle_file inputs
@@ -38,9 +39,9 @@ def virtual_tryon(
         scale=2.5,
         seed=42,
         vt_model_type="viton_hd",
-        vt_garment_type="upper_body",
+        vt_garment_type=garment_type,
         vt_repaint=False,
-        api_name="/leffa_predict_vt")
+        api_name="/leffa_predict_vt",)
     # result[0] is the generated image filepath on the server
     return result[0]  # Gradio will download & display this file
 
@@ -54,11 +55,11 @@ with gr.Blocks() as demo:
         ref = gr.Image(sources="upload",
                        type="filepath",
                        label="Garment Image")
-    vt_garment_type = gr.Radio(
-        label="Garment Type",
+    garment_type = gr.Radio(
         choices=[("Upper", "upper_body"), ("Lower", "lower_body"),
                  ("Dress", "dresses")],
         value="upper_body",
+        label="Garment Type",
     )
 
     with gr.Column():
@@ -67,7 +68,7 @@ with gr.Blocks() as demo:
             label="Result",
         )
     btn = gr.Button("Generate")
-    btn.click(virtual_tryon, [src, ref, vt_garment_type], out)
+    btn.click(virtual_tryon, [src, ref, garment_type], out)
 
 demo.launch(
     share=True,