Merge pull request #4 from fAIseh00d/main

Fix output mask redundant dims

birefnetNode.py

@@ -199,7 +199,7 @@ def rem_bg(self, model, images):
             image = apply_mask_to_image(image.cpu(), mask.cpu())
 
             _images.append(image)
-            _masks.append(mask)
+            _masks.append(mask.squeeze(0))
 
         out_images = torch.cat(_images, dim=0)
         out_masks = torch.cat(_masks, dim=0)