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