BiRefNet / README.md
not-lain's picture
Push model using huggingface_hub.
4d099e5 verified
|
Raw
History Blame Contribute Delete
785 Bytes
metadata
library_name: nobg
tags:
  - birefnet
  - model_hub_mixin
  - nobg
  - pytorch_model_hub_mixin

This model has been pushed to the Hub using the PytorchModelHubMixin integration.

Library: nobg

how to load

pip install nobg

use the AutoModel class

from nobg import AutoModel
model = AutoModel.from_pretrained("nobg/BiRefNet")

or you can use the model class directly

from nobg import BiRefNet
model = BiRefNet.from_pretrained("nobg/BiRefNet")

Contributions

Any contributions are welcome at https://github.com/feyninc/nobg