summaryrefslogtreecommitdiff
path: root/candle-wasm-examples/segment-anything/src
diff options
context:
space:
mode:
Diffstat (limited to 'candle-wasm-examples/segment-anything/src')
-rw-r--r--candle-wasm-examples/segment-anything/src/bin/m.rs5
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 })?)
}
}