automatically load the model with huggingface

README.md

@@ -30,7 +30,7 @@ general stock images, e-commerce, gaming, and advertising content, making it sui
 The accuracy, efficiency, and versatility currently rival leading source-available models. 
 It is ideal where content safety, legally licensed datasets, and bias mitigation are paramount. 
 
-Developed by BRIA AI, RMBG v1.4 is available as a source-available model for non-commercial use. 
+Developed by BRIA AI, RMBG v1.4 is available as an source-available model for non-commercial use. 
 
 [CLICK HERE FOR A DEMO](https://huggingface.co/spaces/briaai/BRIA-RMBG-1.4)
 ![examples](t4.png)
@@ -40,7 +40,7 @@ Developed by BRIA AI, RMBG v1.4 is available as a source-available model for non
 - **Developed by:** [BRIA AI](https://bria.ai/)
 - **Model type:** Background Removal 
 - **License:** [bria-rmbg-1.4](https://bria.ai/bria-huggingface-model-license-agreement/)
-  - The model is released under a Creative Commons license for non-commercial use.
+  - The model is released under an Creative Commons license for non-commercial use.
   - Commercial use is subject to a commercial agreement with BRIA. [Contact Us](https://bria.ai/contact-us) for more information. 
 
 - **Model Description:** BRIA RMBG 1.4 is a saliency segmentation model trained exclusively on a professional-grade dataset.

config.json

@@ -0,0 +1,25 @@
+{
+  "_name_or_path": "not-lain/CustomCodeForRMBG",
+  "architectures": [
+    "BriaRMBG"
+  ],
+  "auto_map": {
+    "AutoConfig": "not-lain/CustomCodeForRMBG--MyConfig.RMBGConfig",
+    "AutoModelForImageSegmentation": "not-lain/CustomCodeForRMBG--briarmbg.BriaRMBG"
+  },
+  "custom_pipelines": {
+    "image-segmentation": {
+      "impl": "not-lain/CustomCodeForRMBG--MyPipe.RMBGPipe",
+      "pt": [
+        "AutoModelForImageSegmentation"
+      ],
+      "tf": [],
+      "type": "image"
+    }
+  },
+  "in_ch": 3,
+  "model_type": "SegformerForSemanticSegmentation",
+  "out_ch": 1,
+  "torch_dtype": "float32",
+  "transformers_version": "4.38.0.dev0"
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:46ef7fe46f2ae284d8f1aaa24bfa5fca5ef25a34e2c7caa890a0029eb100e87f
+size 176381984