1This example project demonstrates using the `wasi-nn` API to perform PyTorch based inference. It consists of Rust code that is built using the `wasm32-wasip1` target.
2
3To run this example:
41. Ensure you set appropriate Libtorch environment variables according to [tch-rs instructions]( https://github.com/LaurentMazare/tch-rs?tab=readme-ov-file#libtorch-manual-install).
5    - Requires the C++ PyTorch library (libtorch) in version *v2.4.0* to be available on
6your system.
7    - `export LIBTORCH=/path/to/libtorch`
82. Build Wasmtime  with `wasmtime-wasi-nn/pytorch` feature.
93. Navigate to this example directory `crates/wasi-nn/examples/classification-example-pytorch`.
104. Download `squeezenet1_1.pt` model
11```
12curl https://github.com/rahulchaphalkar/libtorch-models/releases/download/v0.1/squeezenet1_1.pt --output fixture/model.pt -L
13```
144. Build this example `cargo build --target=wasm32-wasip1`.
155. Run the generated wasm file with wasmtime after mapping the directory containing squeezenet1.1 `model.pt` and sample image `kitten.png`
16    ```
17    ${Wasmtime_root_dir}/target/debug/wasmtime -S nn --dir ${Wasmtime_root_dir}/crates/wasi-nn/examples/classification-example-pytorch::. ${Wasmtime_root_dir}/crates/wasi-nn/examples/classification-example-pytorch/target/wasm32-wasip1/debug/wasi-nn-example-pytorch.wasm
18    ```
196. Check that result `281` has highest probability, which corresponds to `tabby cat`.
20