schirrmacher commited on
Commit
79d0705
·
verified ·
1 Parent(s): 1bb01ef

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. app.py +2 -5
  2. ormbg.pth +2 -2
app.py CHANGED
@@ -60,16 +60,13 @@ def inference(image):
60
  # Gradio interface setup
61
  title = "Open Remove Background Model (ormbg)"
62
  description = r"""
63
- This model is a <strong>fully open-source background remover</strong> optimized for images with humans. It is based on [Highly Accurate Dichotomous Image Segmentation research](https://github.com/xuebinqin/DIS). The model was trained with the synthetic [Human Segmentation Dataset](https://huggingface.co/datasets/schirrmacher/humans), a dataset crafted with [LayerDiffuse](https://github.com/layerdiffusion/LayerDiffuse) and [IC-Light](https://github.com/lllyasviel/IC-Light).
64
 
65
- <img src="file/explanation.jpg" style="width:100%;height:auto;">
66
-
67
- This is the first iteration of the model, so there will be improvements!
68
  If you identify cases where the model fails, <a href='https://huggingface.co/schirrmacher/ormbg/discussions' target='_blank'>upload your examples</a>!
69
 
70
  - <a href='https://huggingface.co/schirrmacher/ormbg' target='_blank'>Model card</a>: find inference code, training information, tutorials
71
  - <a href='https://huggingface.co/schirrmacher/ormbg' target='_blank'>Dataset</a>: see training images, segmentation data, backgrounds
72
- - <a href='https://discord.gg/YYZ3D66t' target='_blank'>Research</a>: join our Discord group!
73
  """
74
 
75
  examples = ["./example1.jpeg", "./example2.jpeg", "./example3.jpeg"]
 
60
  # Gradio interface setup
61
  title = "Open Remove Background Model (ormbg)"
62
  description = r"""
63
+ This model is a <strong>fully open-source background remover</strong> optimized for images with humans. It is based on [Highly Accurate Dichotomous Image Segmentation research](https://github.com/xuebinqin/DIS). The model was trained with the synthetic <a href="https://huggingface.co/datasets/schirrmacher/humans">Human Segmentation Dataset</a>, <a href="https://paperswithcode.com/dataset/p3m-10k">P3M-10k</a> and <a href="https://paperswithcode.com/dataset/aim-500">AIM-500</a>.
64
 
 
 
 
65
  If you identify cases where the model fails, <a href='https://huggingface.co/schirrmacher/ormbg/discussions' target='_blank'>upload your examples</a>!
66
 
67
  - <a href='https://huggingface.co/schirrmacher/ormbg' target='_blank'>Model card</a>: find inference code, training information, tutorials
68
  - <a href='https://huggingface.co/schirrmacher/ormbg' target='_blank'>Dataset</a>: see training images, segmentation data, backgrounds
69
+ - <a href='https://huggingface.co/schirrmacher/ormbg\#research' target='_blank'>Research</a>: see current approach for improvements
70
  """
71
 
72
  examples = ["./example1.jpeg", "./example2.jpeg", "./example3.jpeg"]
ormbg.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2ac115afba3eb111e6cf0bf331ddbd8c14abb4a6ee9695656b9692a549b07f86
3
- size 176717548
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba5817f4d73b494e60d077b4fa2c008c90ad1dc1eb5a7234a958fb0a699907c2
3
+ size 176720018