hotfix

app.py

@@ -5,9 +5,9 @@ from model import SegmentationTool
 seg_tool = SegmentationTool()
 
 
-def segment(input_img, masked_items):
+def segment(input_img, mask_items):
     mask_image, transparent_mask_image, image, items, room = (
-        seg_tool.get_mask(image=input_img, masked_items=masked_items))
+        seg_tool.get_mask(image=input_img, mask_items=mask_items))
     return mask_image
 
 
@@ -16,7 +16,7 @@ demo = gr.Interface(fn=segment,
                         gr.Image(type='pil'),
                         gr.CheckboxGroup([("Door", 14), ("Window", 8)],
                                          value=[8, 14],
-                                         label="Masked Items")],
+                                         label="Mask Items")],
                     outputs=['image'],
                     allow_flagging='never')
 if __name__ == "__main__":

model.py

@@ -68,7 +68,7 @@ class SegmentationTool:
 
         return mask_image
 
-    def get_mask(self, image_path=None, image=None, masked_items=None):
+    def get_mask(self, image_path=None, image=None, mask_items=None):
         if image_path:
             image = Image.open(image_path)
         else:
@@ -83,8 +83,8 @@ class SegmentationTool:
 
         # mask_items = [0, 3, 5, 8, 14]
         # mask_items = [8] # windowpane
-        if masked_items is None:
-            masked_items = []
+        if mask_items is None:
+            mask_items = []
         if 73 in label_ids or 50 in label_ids or 61 in label_ids:
             # mask_items = [0, 3, 5, 8, 14, 50, 61, 71, 73, 118, 124, 129]
             room = 'kitchen'