Xenova HF staff commited on
Commit
8c2f635
·
1 Parent(s): 8ee7c0c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -5
README.md CHANGED
@@ -17,7 +17,7 @@ import { AutoProcessor, VitMatteForImageMatting, RawImage } from '@xenova/transf
17
 
18
  // Load processor and model
19
  const processor = await AutoProcessor.from_pretrained('Xenova/vitmatte-small-distinctions-646');
20
- const model = await VitMatteForImageMatting.from_pretrained('Xenova/vitmatte-small-distinctions-646');
21
 
22
  // Load image and trimap
23
  const image = await RawImage.fromURL('https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/vitmatte_image.png');
@@ -68,10 +68,9 @@ Example inputs:
68
  |--------|--------|
69
  | ![vitmatte_image](https://cdn-uploads.huggingface.co/production/uploads/61b253b7ac5ecaae3d1efe0c/gIwr_finBzqCrzD8Y0Ghm.png) | ![vitmatte_trimap](https://cdn-uploads.huggingface.co/production/uploads/61b253b7ac5ecaae3d1efe0c/ozO5KfIuA3kVZChMelrAZ.png) |
70
 
71
- Example outputs:
72
- | Quantized | Unquantized |
73
- |--------|--------|
74
- | ![output_quantized](https://cdn-uploads.huggingface.co/production/uploads/61b253b7ac5ecaae3d1efe0c/gdfm02xglIi4ZpEIdHg3q.png) | ![output_unquantized](https://cdn-uploads.huggingface.co/production/uploads/61b253b7ac5ecaae3d1efe0c/R57C5wYoCNGn-izrNYeCd.png) |
75
 
76
  ---
77
 
 
17
 
18
  // Load processor and model
19
  const processor = await AutoProcessor.from_pretrained('Xenova/vitmatte-small-distinctions-646');
20
+ const model = await VitMatteForImageMatting.from_pretrained('Xenova/vitmatte-small-distinctions-646', { quantized: false });
21
 
22
  // Load image and trimap
23
  const image = await RawImage.fromURL('https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/vitmatte_image.png');
 
68
  |--------|--------|
69
  | ![vitmatte_image](https://cdn-uploads.huggingface.co/production/uploads/61b253b7ac5ecaae3d1efe0c/gIwr_finBzqCrzD8Y0Ghm.png) | ![vitmatte_trimap](https://cdn-uploads.huggingface.co/production/uploads/61b253b7ac5ecaae3d1efe0c/ozO5KfIuA3kVZChMelrAZ.png) |
70
 
71
+ Example output:
72
+
73
+ ![output_unquantized](https://cdn-uploads.huggingface.co/production/uploads/61b253b7ac5ecaae3d1efe0c/R57C5wYoCNGn-izrNYeCd.png)
 
74
 
75
  ---
76