Update app.py

app.py

@@ -1,8 +1,9 @@
 import gradio as gr
 from SegCloth import segment_clothing
 
-# Define items for each category
+# Define items for each category, including Background
 CATEGORY_ITEMS = {
+    "Background": ["Background"],
     "Clothes": ["Hat", "Upper-clothes", "Skirt", "Pants", "Dress", "Belt", "Left-shoe", "Right-shoe", "Scarf"],
     "Face": ["Face", "Sunglasses"],
     "Hair": ["Hair"],
@@ -22,13 +23,13 @@ iface = gr.Interface(
     inputs=[
         gr.Image(type='pil', label='Image'),
         
-        # Category Selection CheckboxGroup
-        gr.CheckboxGroup(choices=["Clothes", "Face", "Hair", "Skin"],
+        # Category Selection CheckboxGroup including Background
+        gr.CheckboxGroup(choices=["Background", "Clothes", "Face", "Hair", "Skin"],
                          label='Select Categories',
-                         value=["Clothes", "Face", "Hair", "Skin"])  # Default selection
+                         value=["Background", "Clothes", "Face", "Hair", "Skin"])  # Default selection
     ], 
     outputs=gr.Image(label='Clothing Crop'),
-    examples=[['./1.jpg', ["Clothes", "Face"]]],
+    examples=[['./1.jpg', ["Background", "Clothes", "Face"]]],
     theme="Nymbo/Alyx_Theme"
 )