diff options
| author | Radamés Ajna <radamajna@gmail.com> | 2023-09-15 22:34:13 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-16 07:34:13 +0200 |
| commit | 04ca2b9ebd75b065641512d4254850e065699030 (patch) | |
| tree | 7248dee1357db00e748027da7c6be40bbb8faae7 /candle-wasm-examples/segment-anything/src | |
| parent | 635012d770a75033081008a22044804d277fafa8 (diff) | |
| download | candle-04ca2b9ebd75b065641512d4254850e065699030.tar.gz candle-04ca2b9ebd75b065641512d4254850e065699030.tar.bz2 candle-04ca2b9ebd75b065641512d4254850e065699030.zip | |
Update README + SAM (#866)
* use serde-wasm-bindgen, faster serialization
* update readme with demos
Diffstat (limited to 'candle-wasm-examples/segment-anything/src')
| -rw-r--r-- | candle-wasm-examples/segment-anything/src/bin/m.rs | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/candle-wasm-examples/segment-anything/src/bin/m.rs b/candle-wasm-examples/segment-anything/src/bin/m.rs index 949c18a0..5140b979 100644 --- a/candle-wasm-examples/segment-anything/src/bin/m.rs +++ b/candle-wasm-examples/segment-anything/src/bin/m.rs @@ -76,7 +76,7 @@ impl Model { } // x and y have to be between 0 and 1 - pub fn mask_for_point(&self, x: f64, y: f64) -> Result<String, JsError> { + pub fn mask_for_point(&self, x: f64, y: f64) -> Result<JsValue, JsError> { if !(0. ..=1.).contains(&x) { Err(JsError::new(&format!( "x has to be between 0 and 1, got {x}" @@ -112,8 +112,7 @@ impl Model { width: embeddings.width, height: embeddings.height, }; - let json = serde_json::to_string(&MaskImage { mask, image })?; - Ok(json) + Ok(serde_wasm_bindgen::to_value(&MaskImage { mask, image })?) } } |