From 37adb74adf9d7c3b51937a61d4001d91342cd2ee Mon Sep 17 00:00:00 2001
From: uttarayan21 <email@uttarayan.me>
Date: Tue, 19 Aug 2025 18:35:08 +0530
Subject: [PATCH] feat: Save tensors to database as safetensor

---
 src/database.rs | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

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