feat(zkvms/zkm): Implement proper handling of public and private inputs

1 files changed, 6 insertions, 1 deletions

diff --git a/zkvms/zkm/host/src/main.rs b/zkvms/zkm/host/src/main.rs
index 24b4e92..936e981 100644
--- a/zkvms/zkm/host/src/main.rs
+++ b/zkvms/zkm/host/src/main.rs
@@ -93,13 +93,18 @@ async fn main() -> Result<()> {
 
     // Set input
     let mut public_inputstream = Vec::new();
-    bincode::serialize_into(&mut public_inputstream, &run_info.input)
+    bincode::serialize_into(&mut public_inputstream, &run_info.public_input)
+        .expect("public_input serialization failed");
+
+    let mut private_inputstream = Vec::new();
+    bincode::serialize_into(&mut private_inputstream, &run_info.private_input)
         .expect("private_input serialization failed");
 
     let mut prover_input = ProverInput {
         elf: read(elf_path).unwrap(),
         seg_size,
         public_inputstream,
+        private_inputstream,
         ..Default::default()
     };