Update README.md

README.md

@@ -1,18 +1,20 @@
----
-license: mit
-pipeline_tag: mask-generation
-library_name: refiners
-tags:
-- vision
-- image-segmentation
-- matting
-- remove background
-- background
-- background-removal
-- salient-object-detection
-- PyTorch
-- refiners
----
+---
+license: mit
+pipeline_tag: mask-generation
+library_name: refiners
+tags:
+- vision
+- image-segmentation
+- matting
+- remove background
+- background
+- background-removal
+- salient-object-detection
+- PyTorch
+- refiners
+datasets:
+- finegrain/finegrain-product-masks-lite
+---
 
 # Release note for Finegrain Box Segmenter v0.1
 
@@ -170,4 +172,10 @@ mask.save("output.png")
 
 ### With Comfy UI
 
-Coming soon
+Install the `comfyui-refiners` custom nodes:
+```
+comfy node registry-install comfyui-refiners
+```
+
+[Example workflow](https://github.com/finegrain-ai/refiners/blob/main/src/comfyui-refiners/assets/box_segmenter.json)
+[![Workflow](https://raw.githubusercontent.com/finegrain-ai/refiners/main/src/comfyui-refiners/assets/box_segmenter.png)](https://github.com/finegrain-ai/refiners/blob/main/src/comfyui-refiners/assets/box_segmenter.json)