Update app.py

app.py

@@ -163,12 +163,11 @@ def submit_function(
     else:
         mask = automasker(
             person_image,
-            cloth_type
+            cloth_type  # This will now include 'bags' and 'footwear'
         )['mask']
     mask = mask_processor.blur(mask, blur_factor=9)
 
     # Inference
-    # try:
     result_image = pipeline(
         image=person_image,
         condition_image=cloth_image,
@@ -177,10 +176,6 @@ def submit_function(
         guidance_scale=guidance_scale,
         generator=generator
     )[0]
-    # except Exception as e:
-    #     raise gr.Error(
-    #         "An error occurred. Please try again later: {}".format(e)
-    #     )
     
     # Post-process
     masked_person = vis_mask(person_image, mask)
@@ -195,7 +190,7 @@ def submit_function(
             conditions = image_grid([person_image, cloth_image], 2, 1)
         else:
             condition_width = width // 3
-            conditions = image_grid([person_image, masked_person , cloth_image], 3, 1)
+            conditions = image_grid([person_image, masked_person, cloth_image], 3, 1)
         conditions = conditions.resize((condition_width, height), Image.NEAREST)
         new_result_image = Image.new("RGB", (width + condition_width + 6, height))
         new_result_image.paste(conditions, (0, 0))
@@ -241,7 +236,7 @@ def app_gradio():
                         )
                         cloth_type = gr.Radio(
                             label="Try-On Cloth Type",
-                            choices=["upper", "lower", "overall"],
+                            choices=["upper", "lower", "overall", "bags", "footwear"],
                             value="upper",
                         )