Update README.md
Browse files
README.md
CHANGED
@@ -16,7 +16,7 @@ import { AutoModel, AutoProcessor, RawImage } from '@huggingface/transformers';
|
|
16 |
|
17 |
// Load model and processor
|
18 |
const model_id = 'onnx-community/mediapipe_selfie_segmentation';
|
19 |
-
const model = await AutoModel.from_pretrained(model_id, { dtype:
|
20 |
const processor = await AutoProcessor.from_pretrained(model_id);
|
21 |
|
22 |
// Load image from URL
|
|
|
16 |
|
17 |
// Load model and processor
|
18 |
const model_id = 'onnx-community/mediapipe_selfie_segmentation';
|
19 |
+
const model = await AutoModel.from_pretrained(model_id, { dtype: 'fp32' });
|
20 |
const processor = await AutoProcessor.from_pretrained(model_id);
|
21 |
|
22 |
// Load image from URL
|