Upload folder using huggingface_hub

README.md

@@ -17,6 +17,9 @@ The pipeline requires the following files and directories to exist:
     - `segmentation/model/U2NET.pth` : Pre-trained U2NET model.
     - `sam_vit_h_4b8939.pth` : SAM (Segment Anything Model) checkpoint (ViT-H).
 
+3.  **Code Modules**:
+    - `segmentation/data_loader.py` : Data loading utilities for U2NET.
+
 ### Output
 Results are saved in:
 - `datasets/seg_train/` (U2NET masks)