feat: Save tensors to database as safetensor

2025-08-19 18:35:08 +05:30
parent 47218fa696
commit 37adb74adf
1 changed files with 5 additions and 3 deletions
--- 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