mirror of
https://github.com/huggingface/candle.git
synced 2025-06-19 19:58:35 +00:00
Add a wasm module for the segment anything example. (#797)
This commit is contained in:
19
candle-wasm-examples/segment-anything/src/lib.rs
Normal file
19
candle-wasm-examples/segment-anything/src/lib.rs
Normal file
@ -0,0 +1,19 @@
|
||||
use candle_transformers::models::segment_anything::sam;
|
||||
use wasm_bindgen::prelude::*;
|
||||
|
||||
pub use sam::{Sam, IMAGE_SIZE};
|
||||
|
||||
#[wasm_bindgen]
|
||||
extern "C" {
|
||||
// Use `js_namespace` here to bind `console.log(..)` instead of just
|
||||
// `log(..)`
|
||||
#[wasm_bindgen(js_namespace = console)]
|
||||
pub fn log(s: &str);
|
||||
}
|
||||
|
||||
#[macro_export]
|
||||
macro_rules! console_log {
|
||||
// Note that this is using the `log` function imported above during
|
||||
// `bare_bones`
|
||||
($($t:tt)*) => ($crate::log(&format_args!($($t)*).to_string()))
|
||||
}
|
Reference in New Issue
Block a user