1 files changed, 14 insertions, 0 deletions
diff --git a/candle-onnx/src/lib.rs b/candle-onnx/src/lib.rs
new file mode 100644
index 00000000..3b36c4cf
--- /dev/null
+++ b/candle-onnx/src/lib.rs
@@ -0,0 +1,14 @@
+use candle::Result;
+use prost::Message;
+
+pub mod onnx {
+    include!(concat!(env!("OUT_DIR"), "/onnx.rs"));
+}
+
+mod eval;
+pub use eval::simple_eval;
+
+pub fn read_file<P: AsRef<std::path::Path>>(p: P) -> Result<onnx::ModelProto> {
+    let buf = std::fs::read(p)?;
+    onnx::ModelProto::decode(buf.as_slice()).map_err(candle::Error::wrap)
+}