diff --git a/src/database.rs b/src/database.rs
index 2fef122..43e356b 100644
--- a/src/database.rs
+++ b/src/database.rs
@@ -308,9 +308,11 @@ impl FaceDatabase {
         embedding: ndarray::ArrayView1<f32>,
         model_name: &str,
     ) -> Result<i64> {
-        // Convert f32 slice to bytes
-        // let embedding_bytes: Vec<u8> = embedding.iter().flat_map(|&f| f.to_le_bytes()).collect();
-        let embedding_bytes = ndarray_safetensors::SafeArrays::new();
+        let embedding_bytes =
+            ndarray_safetensors::SafeArrays::from_ndarrays([("embedding", embedding)])
+                .change_context(Error)?
+                .serialize()
+                .change_context(Error)?;
 
         let mut stmt = self
             .conn