not-lain commited on
Commit
72dcdab
1 Parent(s): d2056e8

fix config

Browse files
Files changed (2) hide show
  1. README.md +6 -7
  2. config.json +0 -18
README.md CHANGED
@@ -1,17 +1,16 @@
1
  ---
2
  library_name: birefnet
3
  tags:
4
- - background-removal
5
- - mask-generation
 
 
6
  - Dichotomous Image Segmentation
7
  - Camouflaged Object Detection
8
  - Salient Object Detection
9
  - pytorch_model_hub_mixin
10
- - model_hub_mixin
11
- repo_url: https://github.com/ZhengPeng7/BiRefNet
12
- pipeline_tag: image-segmentation
13
  ---
14
 
15
- This model has been pushed to the Hub using **birefnet**:
16
- - Repo: https://github.com/ZhengPeng7/BiRefNet
17
  - Docs: [More Information Needed]
 
1
  ---
2
  library_name: birefnet
3
  tags:
4
+ - model_hub_mixin
5
+ - Image Segmentation
6
+ - Background Removal
7
+ - Mask Generation
8
  - Dichotomous Image Segmentation
9
  - Camouflaged Object Detection
10
  - Salient Object Detection
11
  - pytorch_model_hub_mixin
 
 
 
12
  ---
13
 
14
+ This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
15
+ - Library: https://github.com/ZhengPeng7/BiRefNet
16
  - Docs: [More Information Needed]
config.json CHANGED
@@ -1,21 +1,3 @@
1
  {
2
- "_name_or_path": "ZhengPeng7/BiRefNet",
3
- "architectures": [
4
- "BiRefNet"
5
- ],
6
- "auto_map": {
7
- "AutoConfig": "BiRefNet_config.BiRefNetConfig",
8
- "AutoModelForImageSegmentation": "birefnet.BiRefNet"
9
- },
10
- "custom_pipelines": {
11
- "image-segmentation": {
12
- "impl": "BiRefNet_pipe.BiRefNetPipe",
13
- "pt": [
14
- "AutoModelForImageSegmentation"
15
- ],
16
- "tf": [],
17
- "type": "image"
18
- }
19
- },
20
  "bb_pretrained": false
21
  }
 
1
  {
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  "bb_pretrained": false
3
  }