Update app.py

app.py

@@ -10,13 +10,13 @@ CATEGORY_ITEMS = {
     "Skin": ["Left-arm", "Right-arm", "Left-leg", "Right-leg"]
 }
 
-def segment(img, selected_categories):
+def segment(image, categories_to_hide):
     # Get the items based on the selected categories
     selected_items = []
-    for category in selected_categories:
+    for category in categories_to_hide:
         selected_items.extend(CATEGORY_ITEMS.get(category, []))
     
-    return segment_clothing(img, selected_items)
+    return segment_clothing(image, selected_items)
 
 iface = gr.Interface(
     fn=segment,