Update README.md
Browse files
README.md
CHANGED
@@ -44,7 +44,7 @@ You can then use the model for image matting, as follows:
|
|
44 |
import { AutoModel, AutoProcessor, RawImage } from '@huggingface/transformers';
|
45 |
|
46 |
// Load model and processor
|
47 |
-
const model_id = 'onnx-community/
|
48 |
const model = await AutoModel.from_pretrained(model_id, { dtype: 'fp32' });
|
49 |
const processor = await AutoProcessor.from_pretrained(model_id);
|
50 |
|
|
|
44 |
import { AutoModel, AutoProcessor, RawImage } from '@huggingface/transformers';
|
45 |
|
46 |
// Load model and processor
|
47 |
+
const model_id = 'onnx-community/BiRefNet_lite';
|
48 |
const model = await AutoModel.from_pretrained(model_id, { dtype: 'fp32' });
|
49 |
const processor = await AutoProcessor.from_pretrained(model_id);
|
50 |
|