background-removal-js

background-removal-js copied to clipboard

First of all, really amazing demo! The models produce high quality results - and also quite quickly (considering that it is being run in the browser). Just a few questions:

1. I see you have the model weights stored in the repo; would you like to also add them to the Hugging Face Hub?
2. Which model architecture are you using? If it's compatible with Hugging Face transformers, we'd love to add it to Transformers.js (docs)! Is that something you'd be interested in?
3. Have you experimented with onnxruntime-web's WebGPU backend?

Hi @xenova,

1. Appreciate the offer, can you let me know what are the benefits for the library users? Easier distribution of the model? Currently, I am serving the model via unpkg.com
2. I have to check next week how it could fit into transformers.js
3. I have, however, there were some issues getting the model to work. For some reason the model threw errors after loading. I am assuming some not implemented opcodes, however, this is one of the next things on my list.

Hi! 👋

1. The Hugging Face Hub is the go-to place for sharing your machine learning projects: models, datasets, apps, you name it! There are currently over 240k open-source models already on the hub, and the tagging functionality will definitely help users find your model more easily. For example, there are only 151 image-to-image models (mostly stable diffusion / control nets), so yours could rank quite highly!
2. What architecture does it use? Hugging Face transformers currently supports 206 different architectures, so, if it's one of those, it will be simple to make it compatible! And even if it's not, there are ways to run custom models :)
3. Same problems on my side too! I've been experimenting, but unfortunately I always run into issues (mostly missing ops) when trying to use it. Hopefully one day we can get it working!