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
|